Depending on the shape of landmarks Normalize might not work.
In your code you are assuming to normalize image tensors with 3 channels (thus the stats contain 3 values), which I don’t think is your use case.
Calculate the mean and std (or min and max) from all training targets and normalize the landmarks tensors with them manually.