@ptrblck, in the encoder when I call view. I’m actually go from a tensor of [64,256,3,3] to [576,256]. Then I get as I go to call the encoder 576 stays as the actual batch size which messes up my final dimensionality. I get the error input and target shapes do not match. Being that the only difference is the batch.
Do you know why the View function is doing this? And how to keep the batch size constant?
Thanks