search

Tramac / awesome-semantic-segmentation-pytorch

Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
Apache License 2.0
2.79k stars 580 forks source link

Storage has wong size #170

Closed liu-zg15 closed 3 years ago

liu-zg15 commented 3 years ago

Thank you for your repo. I trained a fcn_resnet50 with 2GPUS. Batch size is 8. When I want to eval the model. I used the eval(ditributed). But get RunTimeERROR: storage has wrong size: expected -4807629896541614912 got 262144. Why this happened? and how to solve it? @Tramac @roclark @liminn @MaKaNu

Tramac commented 3 years ago

Hi, maybe the batch is too large to OOM.