pass loss parameter to add_training_operators function

parent 53a1e745
Pipeline #105552 failed with stages
in 1 minute