I think the problem is the ToTensor function in MyUtils.

it returned a dict, but the Normalize function only takes tensor as input. In this case, is there anyway I can only assign the ‘image’ tensor in the dictionary to the normalization function?