Make main.py runnable

This commit is contained in:
Yin Li 2020-03-04 22:23:34 -05:00
parent 5fbf447e2d
commit b7b14a37eb

View File

@ -11,3 +11,7 @@ def main():
train.node_worker(args) train.node_worker(args)
elif args.mode == 'test': elif args.mode == 'test':
test.test(args) test.test(args)
if __name__ == '__main__':
main()