facebookresearch / unbiased-teacher

PyTorch code for ICLR 2021 paper Unbiased Teacher for Semi-Supervised Object Detection
https://arxiv.org/abs/2102.09480
MIT License
415 stars 83 forks source link

teacher model does not in eval mode? #50

Open shaoqb opened 3 years ago

shaoqb commented 3 years ago

https://github.com/facebookresearch/unbiased-teacher/blob/main/ubteacher/engine/trainer.py#L286

It seems that your teacher model does not freeze model weight and convert to eval mode?

Divadi commented 3 years ago

I asked a similar question here the other day #47

shaoqb commented 3 years ago

Do you compare the experimental results of eval mode and training mode? I am experimenting on my data set, and there is probably a 2 point difference in results