Я работаю над
https://github.com/mit-han-lab/proxylessnas/tree/e9cb66de5612531576d9194226477467b1b2d885[proxylessNAS][1]
, но когда я оцениваю его по набору данных imag enet, используя следующую команду:
python eval.py --path ‘Your path to imagent’ --arch proxyless_cpu # pytorch ImageNet
это дает мне ошибку ниже
/ home / gashwin1 / tools / compiled_tools / pytorch / aten / src / THCUNN / ClassNLLCriterion.cu: 106: void cunn_ClassNLLCriterion_updateOutput_kernel (Dtype *, Dtype *, Dtype *, long *, int), , int, int, long) [с Dtype = float, Acctype = float]: блок: [0,0,0], поток: [30,0,0] Утверждение t> = 0 && t = 0 && t Traceback (последний вызов был последним): файл «eval.py», строка 152, в lost.update (loss.item (), _input.size (0)) RuntimeError: ошибка CUDA : срабатывание подтверждения на стороне устройства