Fix calculation of divergence_loss()
Closed this issue · 0 comments
aschuh-hf commented
The divergence loss as implemented takes the sum of all first order derivatives.
deepali/src/deepali/losses/functional.py
Lines 1356 to 1358 in ea3c16d
Instead, only the trace of the Jacobian matrix should be computed.