engine.finalize()
if __name__ == "__main__":
try:
better_exchook.install()
if len(sys.argv) <= 1:
for k, v in sorted(globals().items()):
if k.startswith("test_"):
print("-" * 40)
print("Executing: %s" % k)
v()
print("-" * 40)
else:
assert len(sys.argv) >= 2
for arg in sys.argv[1:]:
print("Executing: %s" % arg)
if arg in globals():
globals()[arg]() # assume function and execute
else:
eval(arg) # assume Python code and execute
finally:
session.close()
del session
tf.reset_default_graph()
import threading
if len(list(threading.enumerate())) > 1:
print("Warning, more than one thread at exit:")
better_exchook.dump_all_thread_tracebacks()
def _sig_handler(num, frame):
print("Got signal. Dump threads.")
import better_exchook
better_exchook.dump_all_thread_tracebacks()
