I tried it @ptrblck . But that raises an another error of more than one value is ambigious as shown in below image

Capture934×114 30.8 KB
.
also in def train_epoch It gives the same error for
if target is not None:
            target = (target,)
            loss_inputs += target

if I replace target with (target0,target1).