I am working on https://github.com/mit-han-lab/proxylessnas/tree/e9cb66de5612531576d9194226477467b1b2d885
but when I am evaluating it on imagenet dataset, using below command :–
python eval.py --path ‘Your path to imagent’ --arch proxyless_cpu # pytorch ImageNet
it gives me below error
/home/gashwin1/tools/compiled_tools/pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:106: void cunn_ClassNLLCriterion_updateOutput_kernel(Dtype *, Dtype *, Dtype *, long *, Dtype *, int, int, int, int, long) [with Dtype = float, Acctype = float]: block: [0,0,0], thread: [30,0,0] Assertion  t >= 0 && t < n_classes  failed.
/home/gashwin1/tools/compiled_tools/pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:106: void cunn_ClassNLLCriterion_updateOutput_kernel(Dtype *, Dtype *, Dtype *, long *, Dtype *, int, int, int, int, long) [with Dtype = float, Acctype = float]: block: [0,0,0], thread: [31,0,0] Assertion  t >= 0 && t < n_classes  failed.
Test: 0%| | 0/391 [01:55<?, ?it/s]
Traceback (most recent call last):
File “eval.py”, line 152, in
losses.update(loss.item(), _input.size(0))
RuntimeError: CUDA error: device-side assert triggered
Any thoughts please