Closed mattiasegu closed 11 months ago
This PR excludes specific keys exclude_attributes in the loss_dict from the total_loss computation, allowing to log such keys as metrics that are not optimized.
exclude_attributes
loss_dict
total_loss
This PR excludes specific keys
exclude_attributesin theloss_dictfrom thetotal_losscomputation, allowing to log such keys as metrics that are not optimized.