I read somewhere in the notebook I have this:
data_transform = transforms.Compose([
        transforms.RandomSizedCrop(224),
        transforms.RandomHorizontalFlip(),
        transforms.ToTensor(),
        transforms.Normalize(mean=[0.485, 0.456, 0.406],
                             std=[0.229, 0.224, 0.225])
    ])

So, I am not sure what I set the mean and std to for either

        batch = next(iter(train_loader))
        images, landmarks = batch['image'], batch['landmarks']
        
        images = images.cuda()
    
        landmarks = landmarks.view(landmarks.size(0),-1).cuda() 
        
        images = torchvision.transforms.Normalize(images, mean=[0.485, 0.456, 0.406], 
                                                          std=[0.229, 0.224, 0.225])
        landmarks = torchvision.transforms.Normalize(landmarks)

Also, should I normalize images and landmarks after doing a .cuda on them?
Additionally, how?
This is what I have from landmark:
Screenshot_2020-10-08_02-17-101780×732 83 KB
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-21-88ae116843f0> in <module>
     33 
     34         images = torchvision.transforms.Normalize(images, mean=[0.485, 0.456, 0.406], 
---> 35                                                           std=[0.229, 0.224, 0.225])
     36         landmarks = torchvision.transforms.Normalize(landmarks)
     37 

TypeError: __init__() got multiple values for argument 'mean'