Why is the batch size of labeled data equal to the one of unlabeled data?

yulequan / UA-MT

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

https://arxiv.org/abs/1907.07034

477 stars 97 forks source link

Why is the batch size of labeled data equal to the one of unlabeled data? #10

Closed szhang963 closed 3 years ago

szhang963 commented 3 years ago

Whether can I set the batch size of unlabeled data 3 times the batch size of labeled data? What is different between them?Thank you for your answer !!!

yulequan commented 3 years ago

You can change the batch size and the number of labeled data in each batch via the arguments "batch_size" and "labled_bs".