diff --git a/train.py b/train.py index 5a96d1a8..04122e48 100644 --- a/train.py +++ b/train.py @@ -42,7 +42,7 @@ def main(args): def signal_handler(signal, frame): print(" !! Pressed Ctrl+C !!") remove_experiment_folder(OUT_PATH) - sys.exit(0) + sys.exit(1) signal.signal(signal.SIGINT, signal_handler) dataset = LJSpeechDataset(os.path.join(c.data_path, 'metadata.csv'), diff --git a/utils/.generic_utils.py.swp b/utils/.generic_utils.py.swp index a1e05635..5b476405 100644 Binary files a/utils/.generic_utils.py.swp and b/utils/.generic_utils.py.swp differ diff --git a/utils/generic_utils.py b/utils/generic_utils.py index 4a10b9a2..4ca9e632 100644 --- a/utils/generic_utils.py +++ b/utils/generic_utils.py @@ -34,8 +34,9 @@ def remove_experiment_folder(experiment_path): checkpoint_files = glob.glob(experiment_path+"/*.pth.tar") if len(checkpoint_files) < 1: - shutil.rmtree(experiment_path) - print(" ! Run is removed from {}".format(experiment_path)) + if os.path.exists(experiment_path): + shutil.rmtree(experiment_path) + print(" ! Run is removed from {}".format(experiment_path)) else: print(" ! Run is kept in {}".format(experiment_path))