RuntimeError: invalid argument 5: k not in range for dimension at C:/w/b/windows/pytorch/aten/src\THC/generic/THCTensorTopK.cu:24
Opened this issue · 0 comments
Zd-hub commented
Running python run_nn_cls.py --dataset mn40
Error, please ask what the problem is