Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on 4 days ago

Commit

049a0bc

1 Parent(s): 9a0003a

update

Files changed (1) hide show

examples/dfnet/step_2_train_model.py CHANGED Viewed

@@ -168,7 +168,7 @@ def main():
     # optimizer
     logger.info("prepare optimizer, lr_scheduler, loss_fn, evaluation_metric")
-    optimizer = torch.optim.AdamW(model.named_parameters(), config.lr)
     # resume training
     last_step_idx = -1

     # optimizer
     logger.info("prepare optimizer, lr_scheduler, loss_fn, evaluation_metric")
+    optimizer = torch.optim.AdamW(model.parameters(), config.lr)
     # resume training
     last_step_idx = -1