remove folder if error

This commit is contained in:
Eren Golge 2018-04-26 05:44:18 -07:00
parent 379ff17a9a
commit 1aaa78b091
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()