Loss Function In Pytorch Dataloader