From 4c8c3228614038e24cbd3eaa93000c13cec9590c Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 2 Jul 2021 14:58:05 +0200
Subject: [PATCH] Reduce number of filters.
---
climax/main/downscale.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/climax/main/downscale.py b/climax/main/downscale.py
index de87234..85b9bae 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)
--
GitLab