Thanks a lot. Like you mentioned I was able to Normalize images.

        landmarks = landmarks.view(landmarks.size(0),-1).cuda() 
    
        norm = transforms.Normalize([97.13, 97.15, 97.15], [28.74, 28.79, 28.81]) #find the args later
        image = image.float()

        images = norm(images)
        ##landmarks = torchvision.transforms.Normalize(landmarks) #Do I need to normalize the target?
        
        predictions = network(images)