cleaned up pep8 issues

This commit is contained in:
William Falcon
2019-08-07 10:19:03 -04:00
parent 10e4b18452
commit a2d6d514d5
2 changed files with 1 additions and 3 deletions
+1 -1
View File
@@ -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
-2
View File
@@ -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