ShichenLiu / CondenseNet

CondenseNet: Light weighted CNN for mobile devices
MIT License
694 stars 131 forks source link

cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:383 I have only one 2080ti,but I encounter this bug, could you give me some suggestions,please? #23

Open jwnirvana opened 5 years ago

ShichenLiu commented 5 years ago

Hi,

The code should be working on 2080Ti with pytorch 1.0. Could you provide the training command that you used?