mirror of https://github.com/coqui-ai/TTS.git
remove folder if error
This commit is contained in:
parent
c894919120
commit
d143616ada
10
train.py
10
train.py
|
@ -8,6 +8,7 @@ import signal
|
||||||
import argparse
|
import argparse
|
||||||
import importlib
|
import importlib
|
||||||
import pickle
|
import pickle
|
||||||
|
import traceback
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
import torch.nn as nn
|
import torch.nn as nn
|
||||||
|
@ -395,5 +396,10 @@ def main(args):
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
signal.signal(signal.SIGINT, signal_handler)
|
# signal.signal(signal.SIGINT, signal_handler)
|
||||||
main(args)
|
try:
|
||||||
|
main(args)
|
||||||
|
except:
|
||||||
|
signal_handler()
|
||||||
|
traceback.print_exc()
|
||||||
|
sys.exit()
|
||||||
|
|
Loading…
Reference in New Issue