eval running - offline logger implemented -> Test it!

2020-05-30 18:12:42 +02:00
parent ba7c0280ae
commit 8d0577b756
9 changed files with 212 additions and 102 deletions
--- a/main.py
+++ b/main.py
@ -66,8 +66,8 @@ def run_lightning_loop(config_obj):
        trainer.fit(model)

        # Save the last state & all parameters
-        trainer.save_checkpoint(config_obj.exp_path.log_dir / 'weights.ckpt')
-        model.save_to_disk(config_obj.exp_path)
+        trainer.save_checkpoint(logger.log_dir / 'weights.ckpt')
+        model.save_to_disk(logger.log_dir)

        # Evaluate It
        if config_obj.main.eval: