Feeding dataloader batch to model is not working

Also answered here.