change feature size based on 256x256 input

2023-01-18 23:19:53 +00:00
parent 1e1f9e2e09
commit f4341c5e6c
1 changed files with 2 additions and 1 deletions
@@ -26,10 +26,11 @@ class ViewEncoder(nn.Module):
        x = self.act1(x)
        x = self.lin2(x)
        x = self.act2(x)
-        x = torch.ones((x.shape[0], self.features, 15, 20), device=x.get_device()) * x[:,:,None,None]
+        x = torch.ones((x.shape[0], self.features, 8, 8), device=x.get_device()) * x[:,:,None,None]

        return x

+
 from torchvision.ops.misc import Conv2dNormActivation, Permute

 class DumbReconstructor(nn.Module):