`net.train()` should be moved from the main() to the train()

[wenliangdai]

In all of your train.py files, the function call net.train() should be placed in the train function, rather than the main function.

Thanks!