Spaces:

huzey
/

ncut-pytorch

Running on Zero

huzey commited on Sep 24

Commit

f28cf15

•

1 Parent(s): 364ad59

fix batch

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1363,7 +1363,6 @@ def fit_trans(rgb1, rgb2, num_layer=3, width=512, batch_size=256, lr=3e-4, fitti
     # Initialize model and trainer
     mlp = MLP(num_layer=num_layer, width=width, lr=lr, fitting_steps=fitting_steps, seg_loss_lambda=seg_loss_lambda)
     trainer = pl.Trainer(
-        auto_scale_batch_size='power',
         max_epochs=100000,
         gpus=1,
         max_steps=fitting_steps,

     # Initialize model and trainer
     mlp = MLP(num_layer=num_layer, width=width, lr=lr, fitting_steps=fitting_steps, seg_loss_lambda=seg_loss_lambda)
     trainer = pl.Trainer(
         max_epochs=100000,
         gpus=1,
         max_steps=fitting_steps,