Open LLL-YUE opened 1 month ago
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