fix training submission script

2025-12-09 09:19:12 +01:00
parent f2ce282a68
commit c1820d5855
2 changed files with 9 additions and 8 deletions
--- a/run.py
+++ b/run.py
@@ -5,7 +5,11 @@ torch.set_float32_matmul_precision("medium")


 def main():
-    LightningCLI(subclass_mode_data=True, subclass_mode_model=True)
+    LightningCLI(
+        subclass_mode_data=True,
+        subclass_mode_model=True,
+        save_config_kwargs={"overwrite": True},
+    )


 if __name__ == "__main__":