Skip to content
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.

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

Missing switch back to train() mode #6

Open
iamgroot42 opened this issue Dec 10, 2024 · 1 comment
Open

Missing switch back to train() mode #6

iamgroot42 opened this issue Dec 10, 2024 · 1 comment

Comments

@iamgroot42
Copy link

Hello. In the following lines:

self.model.eval()

and

self.model.eval()

The model is set to eval() mode (thus batch-norm statistics are not tracked) but is never reverted to train() as it should be, since the model is still in the process of training.

@KaiyuanZh
Copy link
Owner

Hi @iamgroot42, thank you for reporting this. I have fixed it. Please let me know if you encounter any other issues.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants