pytorch-yolo-v3 icon indicating copy to clipboard operation
pytorch-yolo-v3 copied to clipboard

Published 20 hours ago • verified publisher icon ayooshkathuria

Problems using own weights and CFG

Open WrongEngineer opened this issue 6 years ago • 0 comments

hi ive been trying to use the same method you did but with weights and cfg setup for just one class, however whenever i run it i get the error "Got 104 and 52 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:71". ive tested the weights with openCV-python and darknet itself and it works. any ideas on how to fix the issue?

WrongEngineer avatar Nov 05 '19 10:11 WrongEngineer