Fix LRScheduler and ckpt_path

README.md

@@ -22,7 +22,7 @@ Models and effects detailed below.
 
 To add gpu, add `trainer.accelerator='gpu' trainer.devices=-1` to the command-line
 
-Ex. `python scripts/train.py +exp=umx_distortion trainer.accelerator='gpu' trainer.devices=-1`
+Ex. `python scripts/train.py +exp=umx_distortion trainer.accelerator='gpu' trainer.devices=1`
 
 ### Current Models
 - `umx`

cfg/config.yaml

@@ -77,6 +77,7 @@ trainer:
   enable_model_summary: False
   log_every_n_steps: 1 # Logs metrics every N batches
   accumulate_grad_batches: 1
+  deterministic: True
   accelerator: null
   devices: 1
   gradient_clip_val: 10.0

remfx/models.py

@@ -63,8 +63,9 @@ class RemFXModel(pl.LightningModule):
         optimizer,
         optimizer_idx,
         optimizer_closure,
-        on_tpu=False,
-        using_lbfgs=False,
+        on_tpu,
+        using_native_amp,
+        using_lbfgs,
     ):
         # update params
         optimizer.step(closure=optimizer_closure)

scripts/train.py

@@ -42,7 +42,7 @@ def main(cfg: DictConfig):
     summary = ModelSummary(model)
     print(summary)
     trainer.fit(model=model, datamodule=datamodule)
-    trainer.test(model=model, datamodule=datamodule)
+    trainer.test(model=model, datamodule=datamodule, ckpt_path="best")
 
 
 if __name__ == "__main__":