refactored model tests

2019-07-24 13:42:42 -04:00 · 2019-07-24 13:42:42 -04:00 · 8a43f4307e
parent b90841dc3d
commit 8a43f4307e
1 changed files with 2 additions and 0 deletions
--- a/tests/debug.py
+++ b/tests/debug.py
@ -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],