question: double evaluation during training

facebookresearch / unbiased-teacher

PyTorch code for ICLR 2021 paper Unbiased Teacher for Semi-Supervised Object Detection

https://arxiv.org/abs/2102.09480

MIT License

410 stars 82 forks source link

question: double evaluation during training #25

Closed hachreak closed 3 years ago

hachreak commented 3 years ago

Hi everybody, on test every EVAL_PERIOD iterations start a evaluation. Inside the logs I see that it runs 2 times. Is the first for the teacher model and the second for the student model? Or viceversa? Where do you run the evaluation inside the code? Thanks a lot.

ycliu93 commented 3 years ago

Hi @hachreak ,

Yes, the evaluation runs twice. The first one is for student, and the second one is for teacher.

https://github.com/facebookresearch/unbiased-teacher/blob/ba543ed2b87446c79eaf41f93320de246c8eedc2/ubteacher/engine/trainer.py#L618-L619

This is where we do the evaluations, and we follow the default evaluation function in Detectron2. Thanks!