Question about only supervised training

yyliu01 / PS-MT

[CVPR'22] Perturbed and Strict Mean Teachers for Semi-supervised Semantic Segmentation

https://arxiv.org/pdf/2111.12903.pdf

MIT License

186 stars 17 forks source link

Question about only supervised training #31

Closed abhipsychh closed 1 year ago

abhipsychh commented 1 year ago

Hey, I was trying to train the VOC model using only supervised learning, so I changed supervised = True and semi = False in deeplabv3+ config and only gave a labeled text file but I am not able to train it. What other changes should I do?

yyliu01 commented 1 year ago

Hi @abhij247

Could you please provide more details about the error feedback from the code?

Why didn't you just simply set the unsupervised weight to 0? I know there are some differences about the batch norm layers for the unlabelled data, but it shouldn't be a big deal.

Cheers, Yuyuan

abhipsychh commented 1 year ago

Thanks, I have set the unsupervised weight to 0 for the supervised training.