T5 user defined loss function

Thanks! This guidance is very helpful!