Use checkpoint_connector.hpc_save in SLURM (#4217)

Lightning-AI · Oct 18, 2020 · 66e58f5 · 66e58f5
1 parent 2e9733b
commit 66e58f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/connectors/slurm_connector.py b/pytorch_lightning/trainer/connectors/slurm_connector.py
@@ -85,7 +85,7 @@ def sig_handler(self, signum, frame):  # pragma: no-cover
         if self.trainer.is_global_zero:
             # save weights
             log.info('handling SIGUSR1')
-            self.trainer.hpc_save(self.trainer.weights_save_path, self.trainer.logger)
+            self.trainer.checkpoint_connector.hpc_save(self.trainer.weights_save_path, self.trainer.logger)
 
             # find job id
             job_id = os.environ['SLURM_JOB_ID']