coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix handling CTRL C

pull/10/head
Eren Golge 2018-01-24 08:17:49 -08:00
parent 2a20b7c2ac
commit d95a2ce26f
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -42,7 +42,7 @@ def main(args):
def signal_handler(signal, frame): def signal_handler(signal, frame):
print(" !! Pressed Ctrl+C !!") print(" !! Pressed Ctrl+C !!")
remove_experiment_folder(OUT_PATH) remove_experiment_folder(OUT_PATH)
sys.exit(0) sys.exit(1)
signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGINT, signal_handler)
dataset = LJSpeechDataset(os.path.join(c.data_path, 'metadata.csv'), dataset = LJSpeechDataset(os.path.join(c.data_path, 'metadata.csv'),

BIN
utils/.generic_utils.py.swp
View File

Binary file not shown.

5
utils/generic_utils.py
View File

@ -34,8 +34,9 @@ def remove_experiment_folder(experiment_path):
checkpoint_files = glob.glob(experiment_path+"/*.pth.tar") checkpoint_files = glob.glob(experiment_path+"/*.pth.tar")
if len(checkpoint_files) < 1: if len(checkpoint_files) < 1:
shutil.rmtree(experiment_path) if os.path.exists(experiment_path):
print(" ! Run is removed from {}".format(experiment_path)) shutil.rmtree(experiment_path)
print(" ! Run is removed from {}".format(experiment_path))
else: else:
print(" ! Run is kept in {}".format(experiment_path)) print(" ! Run is kept in {}".format(experiment_path))