dd604/refinedet.pytorch

RuntimeError: invalid argument 2: non-empty vector or matrix expected at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:31

Fuhan1994 opened this issue · 1 comments

when i run the train_refinedet.py, there is a problem:
RuntimeError: invalid argument 2: non-empty vector or matrix expected at /pytorch/aten/src/THCUNN/generic/ClassNLLCriterion.cu:31

dd604 commented

Maybe there is some error about gt and loss. You would better check the dataset and the ground truth.