New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Why does main() train the same model twice? #12

Open

LLL-YUE opened this issue Jul 6, 2024 · 1 comment

LLL-YUE commented Jul 6, 2024

Owner

noelshin commented Jul 6, 2024

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment