pass loss parameter to add_training_operators function

parent 8dffa375
Pipeline #105551 failed with stages
in 1 minute and 2 seconds