Replies: 1 comment
-
I would also like to see this functionality. However you can always look into the code and write a custom function that calculates both losses and inference results. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Very often it is desirable to compute both loss function and different metrics using inference results on a validation set. For this, in a current version of
GeneralizedRCNN
, two methods should be called separately (forward
andinference
) requiring two network passes. Adding inference results to return values will increase validation time when both loss and metrics are computed.Beta Was this translation helpful? Give feedback.
All reactions