Is your feature request related to a problem? Please describe.
Add loss computation for RetinaNet

Describe the solution you'd like
Add loss computation for RetinaNet

Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.

Additional context
Add any other context or screenshots about the feature request here.