do not remove weights after save

Lightning-AI · Aug 14, 2020 · 6b27dc2 · 6b27dc2
1 parent 8f83bd2
commit 6b27dc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/distrib_data_parallel.py b/pytorch_lightning/trainer/distrib_data_parallel.py
@@ -457,7 +457,7 @@ def load_spawn_weights(self, original_model):
             original_model.load_state_dict(loaded_model.state_dict())
 
             # remove ddp weights
-            os.remove(path)
+            # os.remove(path)
 
         return loaded_model