Problems using own weights and CFG

[WrongEngineer]

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?