diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 1b3a9a7a..2dbf46eb 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -261,7 +261,7 @@ class Trainer(TrainerIO): if '.ckpt' in name: epoch = name.split('epoch_')[1] - epoch = int(re.sub('[^0-9]', '' ,epoch)) + epoch = int(re.sub('[^0-9]', '', epoch)) if epoch > last_epoch: last_epoch = epoch diff --git a/tests/test_models.py b/tests/test_models.py index 746f97ef..f6e792fe 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -123,8 +123,6 @@ def test_amp_gpu_ddp(): run_gpu_model_test(trainer_options, model, hparams) - - def test_cpu_slurm_save_load(): """ Verify model save/load/checkpoint on CPU