Make main.py runnable
This commit is contained in:
parent
5fbf447e2d
commit
b7b14a37eb
1 changed files with 4 additions and 0 deletions
|
@ -11,3 +11,7 @@ def main():
|
|||
train.node_worker(args)
|
||||
elif args.mode == 'test':
|
||||
test.test(args)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
|
Loading…
Reference in a new issue