search

LikeLy-Journey / SegmenTron

Support PointRend, Fast_SCNN, HRNet, Deeplabv3_plus(xception, resnet, mobilenet), ContextNet, FPENet, DABNet, EdaNet, ENet, Espnetv2, RefineNet, UNet, DANet, HRNet, DFANet, HardNet, LedNet, OCNet, EncNet, DuNet, CGNet, CCNet, BiSeNet, PSPNet, ICNet, FCN, deeplab)
Apache License 2.0
706 stars 162 forks source link

RuntimeError: stack expects each tensor to be equal size, but got [3, 366, 500] at entry 0 and [3, 335, 500] at entry 1 #58

Open andrew-begain opened 4 years ago

andrew-begain commented 4 years ago

@LikeLy-Journey thanks for your greak work ! when i run tools/eval.py i ran into this problem,and i am use the config of the pascal_voc_deeplabv3_plus.yaml,and i change the BACKBONE: "xception65" to BACKBONE: "resnet50" because of my low gpu memory,so what should i do to fix the problem? thanks

dreamlychina commented 3 years ago

@LikeLy-Journey thanks for your greak work ! when i run tools/eval.py i ran into this problem,and i am use the config of the pascal_voc_deeplabv3_plus.yaml,and i change the BACKBONE: "xception65" to BACKBONE: "resnet50" because of my low gpu memory,so what should i do to fix the problem? thanks

meet the same problem，do you have fixed it ？？