6D prediction files now working

2020-06-26 08:33:59 +02:00
parent 2a7a236b89
commit 358d692699
3 changed files with 5 additions and 5 deletions
--- a/main.py
+++ b/main.py
@ -27,7 +27,7 @@ def run_lightning_loop(config_obj):
        checkpoint_callback = ModelCheckpoint(
            monitor='mean_loss',
            filepath=str(logger.log_dir / 'ckpt_weights'),
-            verbose=True, save_top_k=10,
+            verbose=True, save_top_k=3,
        )

        # =============================================================================