search

durandtibo / wildcat.pytorch

PyTorch implementation of "WILDCAT: Weakly Supervised Learning of Deep ConvNets for Image Classification, Pointwise Localization and Segmentation", CVPR 2017
MIT License
267 stars 61 forks source link

Training stuck at 0% #3

Open Tauranis opened 7 years ago

Tauranis commented 7 years ago

I'm currently using a docker image which include all required pytorch and CUDA 8.0 dependencies. I just cloned the repo and executed the demo_voc2007 with the default argument values. The training stuck at 0%, nothing happens. Is there a known reason why is this happening?

Tauranis commented 7 years ago

This usually happens when the batch size is large, like 32 or Can it 64. Can it be due to the multiprocessing done internally?

Kqiii commented 6 years ago

Hi Tauranis, I encountered the same problem as you. Do you have any solution now?