Make main.py runnable
This commit is contained in:
parent
5fbf447e2d
commit
b7b14a37eb
@ -11,3 +11,7 @@ def main():
|
||||
train.node_worker(args)
|
||||
elif args.mode == 'test':
|
||||
test.test(args)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
Loading…
Reference in New Issue
Block a user