Oh okay - so the width and height of the size array being 256 is fine? Apologies if I’m a bit thick, but would have thought 256 would be a relic of the original model using 256x256 images (and thus be wrong for a 64x64 model).
x.size() returns torch.Size([64, 512, 2, 2]) before I flatten the Tensor in this line: x = x.view(x.size(0),-1) to make its shape/size [64,2048].
I.e. just before passing it to self.fc in the line style = self.fc.forward(x) , x.size() returns  torch.Size([64, 2048]).Does this signal anything?
Also thanks for the tip about calling the Module directly, not sure why I didn’t!