Understanding update() and compute() functions for building custom metric #1192
Unanswered
kkarrancsu
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
I'd like to implement a custom TorchMetric in order to use it as a loss function. A couple of questions:
1 - I noticed there is a property:
is_differentiable
What specifically needs to be done w/ the metric implementation to ensure that the implemented metric is actually differentiable vs. simply just setting the property?The next 2 questions apply to the DDP setting:
a - Is it OK to understand the update() method as the computation that happens on each GPU?
b - Is it OK to understand that the compute() method is what happens when the outputs of update are aggregated across all GPUs?
Beta Was this translation helpful? Give feedback.
All reactions