mxnet-yolo icon indicating copy to clipboard operation
mxnet-yolo copied to clipboard

Loss function

Open shabnamghadar opened this issue 7 years ago • 1 comments

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,

shabnamghadar avatar Oct 13 '17 23:10 shabnamghadar

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.

insanegtp avatar Mar 10 '18 00:03 insanegtp