From f4341c5e6c5f7657b3792e2280e3c4126ff28396 Mon Sep 17 00:00:00 2001
From: Hunor Laczko <luni741@gmail.com>
Date: Wed, 18 Jan 2023 23:19:53 +0000
Subject: [PATCH] change feature size based on 256x256 input

---
 models/depth_estim.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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):