remove folder if error

This commit is contained in:
Eren Golge 2018-04-26 05:44:18 -07:00
parent c894919120
commit d143616ada
1 changed files with 8 additions and 2 deletions

View File

@ -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()