Loss function

[shabnamghadar]

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,

[insanegtp]

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.