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

为什么没有设置训练模式和验证模式呢? #1

Open
shanniruo opened this issue Jul 10, 2020 · 3 comments
Open

为什么没有设置训练模式和验证模式呢? #1

shanniruo opened this issue Jul 10, 2020 · 3 comments

Comments

@shanniruo
Copy link

No description provided.

@yuruiqi
Copy link
Owner

yuruiqi commented Jul 10, 2020

确实如此 该项目正在修改中 之前尚未更新 感谢指正

@shanniruo
Copy link
Author

你好,我最近也在尝试复现mask-rcnn的代码,我不明白在这里为什么训练时要始终保证BN层参数不更新,https://github.com/multimodallearning/pytorch-mask-rcnn/blob/master/model.py#L1627
而且这样做不就是和eval模式一样了吗 我不太理解,请教一下

@yuruiqi
Copy link
Owner

yuruiqi commented Jul 17, 2020

我查阅了一下该项目下的issue,发现了有相同的疑问与作者的解答:https://github.com/multimodallearning/pytorch-mask-rcnn/issues/89。
作者大致意思是由于训练数据图像分辨率较大,因此无法设置大的batchsize,这样便无法有效的训练bn层。
感谢你的问题,对于我的复现也很有帮助。

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