Reduce number of filters.

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)