GuoleiSun / CountSeg

Official code for "Object counting and instance segmentation with image-level supervision", in CVPR 2019 and TPAMI 2020
162 stars 43 forks source link

RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:663 #16

Open wasdzzzzzz opened 2 years ago

wasdzzzzzz commented 2 years ago

My environment: gcc version 4.8.5 CUDA9.0 UBUNTU18.04

Do you know how to solve this problem? Thanks a lot in advance.