search

charlesCXK / TorchSemiSeg

[CVPR 2021] CPS: Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision
MIT License
504 stars 74 forks source link

Question about the batchnorm while trainning #82

Closed AXu0511 closed 1 year ago

AXu0511 commented 1 year ago

Hi Xiaokang, If I have only a GPU, should i change the SyncBatchNorm to nn.BatchNorm2d? Looking forward to your reply. image

charlesCXK commented 1 year ago

Hi, I think it is ok.