wasidennis / AdaptSegNet

Learning to Adapt Structured Output Space for Semantic Segmentation, CVPR 2018 (spotlight)
845 stars 205 forks source link

Is this code applicable to large batch size? #75

Closed Guptajakala closed 4 years ago

Guptajakala commented 4 years ago

You mentioned you are training with batch size 1 in this case. I need to run on a dataset where the images are much smaller and my GPU allows the batch size to be up to 200. Will the adversarial training work in this case?

hfslyc commented 4 years ago

We train with batch size 1 only because of memory limitation. The method ideally works with any batch size. But with different dataset, the hyperparameters or network architecture could be tuned for better performance.