From d95a2ce26fbd319be1b9b6df189405b07ef1c43c Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 24 Jan 2018 08:17:49 -0800 Subject: [PATCH] fix handling CTRL C --- train.py | 2 +- utils/.generic_utils.py.swp | Bin 20480 -> 20480 bytes utils/generic_utils.py | 5 +++-- 3 files changed, 4 insertions(+), 3 deletions(-) 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 a1e0563566e8c234b82fbc063258feead788a996..5b4764055e3bbf304a229c799eb03ce5de898598 100644 GIT binary patch delta 1356 zcmYMzSx8h-9LMqh=s4pt=0ZVP>eX^<84|;w)F`c(N_v=V4^!vLjHP2ap%giiR8&yv z4SI4Rii{T1Au6qW39_gpS|Fr{_H5CLq7t&cw??|~;T+D~d+zZ2-BZ_QsB1IC6wh95 z%U9-F=LlD%$i)lJ^^0TUqW4uyJ#cZ!j>$Vwk2yMaHMu^jo=-{|#dB zkAeQt?jOs2>mtuYgg1tXWLsEccQ#N~dbM#Pb?~AV1&G1paFI4_M+wLn$&kOJk zWkg%FoWu;RBQe@rjnkcwcKp*J#2 z;{E7FE%YO4#Vibx{5=b)cQQE&5A4Xq1pJ_Z0bE55a$tbqCJh|NVH6_~Pe{KRji|yJ zSn-kuum8+Z)eK~EO`~g&A B+|U33 delta 1543 zcmZY9eQXnT9LMqRbsK}NTlXZ9=j#9mO@)R@2!m2!!vG76urXzmY+c*YYiDnaDw-I( z5C}$|92evMVRV5I*&<^Uc~B!yMlo9?8fCy@W;8qo{6is-MHAi!TO#w4SMT?`HrHQ2 z_vx=Wt~AG$@pZ1s6|2;FvtAL7ERjRD`l^p#m@J3Swv6|udVPAdNsn|l`;!uPk_xO! zj**=;WoEVcVUCPvCsBsZNkiwgLuW0qKC9D~;m;Jg1oQsfKrM@fnYp3hHg~y&n>d9K zG&qoo->f40VPG99P!0zM(?xm^#{sm#hhi+n5;%~C!8DOG*pwzQ(HI0V4P$U`jL2nl zVGlN84GJ+9k4B3m@Gcf&0WA16Rpfi@#|Gr#>L`(39K}9puwx)a9nuo6>|jk^|+J2;Q`un3QN4iA);_60`2-DR&1fsjH9jw!YUDb{CA;0K%h=%jHl}j;^hmw7QFW_JqS2-j%@_69 zJ)wYKbGw}CjG0W-8IBr9qqoWXrdr>~(5XI~1R_4&uSGVxSL>0$@I_z4=I7>mT`1s- z)&zAu{M>bJu{ls(Q{SL_qyK-SMvIKNI&!7|S>)42k6h`p#HCpamDq$3>$jWDcxtZFoX&e z;3Yhye1G68Y(*t<@h_$O0%!0M!k7aa&QZKpgs=wru;LfW_cJ>1He8sEdzA0;s9KR9 z7<6J6b|Q?IA?TxwClJ6=*eiP=aEpFjKvs)ZM5$i!6=?tq#2I#i$>)6hq%SMUkm!fK>L!9~)&fb(cY&`Q$t z8Dt^@R{TiP-8hCWwBj8EVLpig67Rzm?86pR!Hr4yhtvn~HQKQa0oZYi#9X-Uq6|bTQ6Cvm$@x53D6`A;k#NS6Xs<0T@_?v`3MJ-lfIbMXhMVv+pVt5^| gVRi~HcP-uRMat8S;5$z~TfnlBfBk|)&{=DF0@7+2)Bpeg 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))