diff --git a/climax/main/downscale.py b/climax/main/downscale.py
index de872342586fc24c79d82eacaad6692299435a14..85b9bae7e24c8057a48ef50f0b052a0bf58d54a8 100644
--- a/climax/main/downscale.py
+++ b/climax/main/downscale.py
@@ -56,7 +56,7 @@ if __name__ == '__main__':
 
     # initialize network
     net = SegNet(MODEL_PATH.joinpath(PREDICTAND + '.pt'),
-                 len(Era5_ds.data_vars), 1, filters=[64, 128, 256])
+                 len(Era5_ds.data_vars), 1, filters=[32, 64, 128])
 
     # initialize optimizer
     optimizer = torch.optim.Adam(net.parameters(), lr=LR)