Chun Han Lu
Chun Han Lu
Apologies for the new spaces, it was automatically modified by Pycharm IDE
`python test.py --dataset ade20k --pretrained --model fcn --backbone resnet50 --eval`  `python test.py --dataset pascal_voc --pretrained --model fcn --backbone resnet101 --eval` 
I faced the same problem while training mnist with WGANGP, this solutions works fine for me. Thanks~
It could be done by modifying with **model = nn.DataParallel(model)** Just by aware that **latency_to_accumulate** has shape [1] which is not allowed to parallel It should be reshape to [#of...