Optimizer Weight Decay Pytorch Dataloader Cuda