I am facing the same issue, I want to disable it altogether because I’m working with single channel images and the transformation gives me 3 channel images even when my input is explicitly given as single channel (model architecture changed prior to that as well). Is there a way to do this without losing the pretrained weights?