search

MeioJane / CHR

SIXray : A Large-scale Security Inspection X-ray Benchmark in CVPR 2019
80 stars 37 forks source link

The relation between outputs of network and the loss function #3

Closed Ao-Lee closed 5 years ago

Ao-Lee commented 5 years ago

The network creates 3 outputs. Is there any additional computation involved between the loss function and the 3 net outputs? Thanks

codestorm04 commented 5 years ago

I guess you can refer to engine.py:116 with the on_forward method, where loss are calculated and o1,o2,o3 are used.

Ao-Lee commented 5 years ago

@codestorm04 thank you !!