search

ShichenLiu / CondenseNet

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

RuntimeError: INDICES element is out of DATA bounds #41

Open MohBarbary opened 2 years ago

MohBarbary commented 2 years ago

Whenever I run the following command:

python main.py --model condensenet_converted -b 64 -j 4 C:\CondenseNet-master --stages 4-6-8-10-8 --growth 8-16-32-64-128 --group-1x1 4 --group-3x3 4 --condense-factor 4 --evaluate-from C:\CondenseNet-master/converted_condensenet_4.pth.tar --gpu 0

I get the following error:

RuntimeError: INDICES element is out of DATA bounds, id=53888868763566084 axis_dim=2064

any idea how to solve this issue?

Thank you in advance.