diff --git a/models/depth_estim.py b/models/depth_estim.py
index f64753a..dc7080a 100644
--- a/models/depth_estim.py
+++ b/models/depth_estim.py
@@ -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):