refactored model tests

2026-09-09 11:32:07 +08:00 · 2019-07-24 13:42:42 -04:00
parent b90841dc3d
commit 8a43f4307e
1 changed files with 2 additions and 0 deletions
@@ -105,6 +105,8 @@ def main():
    checkpoint = ModelCheckpoint(save_dir)

    trainer = Trainer(
+        experiment=exp,
+        checkpoint_callback=checkpoint,
        progress_bar=False,
        max_nb_epochs=1,
        gpus=[0, 1],