Actually using the introduces loss_function_call_batch.
Signed-off-by: Andre Klein <andre.klein@dkfz-heidelberg.de>