Right, I’m trying to feed in a 96X96. the only real difference i made was to upsample at the end. But I’m getting mismatch errors when I try and treat the view in the way you mentioned (specifically in the linear layer just following the view call), i get the error size mismatch, m1: [256 x 576], m2: [256 x 20] at
This is what I’m getting at every layer. Do you see what the problem is? or how I can fix it?
torch.Size([64, 32, 48, 48])
torch.Size([64, 32, 24, 24])
torch.Size([64, 64, 12, 12])
torch.Size([64, 64, 6, 6])
torch.Size([64, 256, 3, 3])
torch.Size([576, 256])
torch.Size([576, 20])
torch.Size([576, 10])
torch.Size([576, 256])
torch.Size([576, 256, 1, 1])
torch.Size([576, 64, 4, 4])
torch.Size([576, 64, 8, 8])
torch.Size([576, 32, 16, 16])
torch.Size([576, 32, 32, 32])
torch.Size([576, 8, 64, 64])