hustvl / BoxTeacher

[CVPR 2023] Exploring High-Quality Pseudo Masks for Weakly Supervised Instance Segmentation
https://arxiv.org/abs/2210.05174
MIT License
75 stars 3 forks source link

AttributeError: 'BoxTeacher' object has no attribute 'module #3

Closed XJWang628 closed 1 year ago

XJWang628 commented 1 year ago

Hi, I encountered "AttributeError: 'BoxTeacher' object has no attribute 'module'" while training BoXTeacher on a single 3090, is there any way to solve it? image

Huster-Hq commented 1 year ago

Hi, my bug is "'FCOS' object has no attribute 'forward_test'".

Huster-Hq commented 1 year ago

嗨,我在单个 3090 上训练 BoXTeacher 时遇到了“属性错误:'BoxTeacher' 对象没有属性'模块'”,有什么办法可以解决吗? 图像 You can find the code where the error occurred and remove the '.module ', I solved the bug with this method。

wondervictor commented 1 year ago

Thank @Huster-Hq for solving this issue. As for that FCOS does not have "forward_test", you might need to use the modified version of FCOS in this repo, in which we have add "forward_test".