I did try to adapt bb_pad by passing in empty string as a label. I’ve also modified the L1Loss to work with multiple outputs. Now I’m getting bool value of Tensor with more than one value is ambiguous error. I’ve posted more about it under similar topic here