Change config to json 3

This commit is contained in:
Eren Golge
2018-01-22 08:29:27 -08:00
parent e225993fb6
commit 1ee45b5336
7 changed files with 11 additions and 9 deletions
+1 -1
View File
@@ -33,7 +33,7 @@ def remove_experiment_folder(experiment_path):
"""Check folder if there is a checkpoint, otherwise remove the folder"""
checkpoint_files = glob.glob(experiment_path+"/*.pth.tar")
if len(checkpoint_files) < 2:
if len(checkpoint_files) < 1:
shutil.rmtree(experiment_path)
print(" ! Run is removed from {}".format(experiment_path))
else: