Make main.py runnable
This commit is contained in:
parent
5fbf447e2d
commit
b7b14a37eb
@ -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()
|
||||||
|
Loading…
Reference in New Issue
Block a user