Yes and i think this is the case. Is it caused by some of the images though?It looks like guesses on those 3 channel tensors are just fine until the error with ‘1 channel one’ occures
image667×579 49.4 KB