yulequan / UA-MT

code for MICCAI 2019 paper 'Uncertainty-aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentation'.

Home Page:https://arxiv.org/abs/1907.07034

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Why is the student model stored instead of a teacher model?

chuxiang93 opened this issue · comments

Dear @yulequan
Another question, why is the student model stored instead of a teacher model in the training phase. In the paper of "Mean Teacher", I found that the teacher model has better performance.
if iter_num % 1000 == 0: save_mode_path = os.path.join(snapshot_path, 'iter_' + str(iter_num) + '.pth') torch.save(model.state_dict(), save_mode_path) logging.info("save model to {}".format(save_mode_path))
Thanks,
Jianqiang Ma

Hi @mjq93, In our experiment. The student model and teacher model achieve similar performance. Therefore, we both use student model for final evaluation.

OK, I know, thank you for your reply.