Why does main() train the same model twice?

[LLL-YUE]

[noelshin]

Hello.

"model.train()" is not the call for training the model. It's to get some functions in the model ready for training (but one of "model.train()" is not necessary there). The actual training takes place at https://github.com/NoelShin/selfmask/blob/master/main.py#L103