Add command line control over whether all loss function components are computed and reported to WandB, independently of whether they are used in the loss function.
Useful for comparing off-target/correlated effects of loss components.
Set to false by default as doing so will add computation time.
Add command line control over whether all loss function components are computed and reported to WandB, independently of whether they are used in the loss function.
Useful for comparing off-target/correlated effects of loss components.
Set to false by default as doing so will add computation time.