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
My environment: gcc version 4.8.5 CUDA9.0 UBUNTU18.04
Do you know how to solve this problem? Thanks a lot in advance.