Fixed the bug in dataloader

MenghaoGuo / AutoDeeplab

Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation

https://arxiv.org/abs/1901.02985

410 stars 97 forks source link

Closed HankKung closed 5 years ago

HankKung commented 5 years ago

train_loader should be returned train_loader twice for alpha and beta training during an archtecture search.