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
Hi,
The code should be working on 2080Ti with pytorch 1.0. Could you provide the training command that you used?