mxnet-yolo
mxnet-yolo copied to clipboard
Loss function
Hi, Can you tell me where did you define the loss function? I want to apply a modification to it and test it out
Thanks,
After my whole day study, I found the loss is defined in the contrib.YoloOutput whose source codes are in the src/operator/contrib/yolo_output-inl.h. And there are other files(.cu .cc) about yolooutput should be considered.