I am not following where you attach the correct set of transforms under training or testing… but it appears you’re calling the dictionary itself, when what you wanted was one of the keys “train” or “val”