ck37/coral-ordinal

OrdinalCrossEntropy does not use `reduction` argument

gmgeorg opened this issue · 0 comments

OrdinalCrossEntropy inherits from baseline Loss class with reduction argument. However, the reduction argument is then ignored in the actual loss computation of the helper function: ordinal_loss() (it uses reduce_mean() even if reduction != tf.keras.losses.Reduction.SUM_OVER_BATCH_SIZE or AUTO)