Open Xinge-Zhou opened 6 months ago
See:
Thanks for your help, I changed the code, it can running now.
"""
Update teacher model weights.
"""
if self.iter == self.model_teacher.burn_in:
self.update_teacher_model(ema_decay=0.)
elif self.iter > self.model_teacher.burn_in:
self.update_teacher_model(ema_decay=self.model_teacher.ema_decay)
I am using coco dataset and already trained the teacher model , command is
python3 -W ignore train_net.py --config-file configs/coco/instance-segmentation/maskformer2_R50_bs16_50ep.yaml --num-gpus 1 --num-machines 1 SSL.PERCENTAGE 250 SSL.TRAIN_SSL False
When I Train semi-supervised model using pretrained checkpoint, I got error as bellow.