From e548799a5615fefa168d84b7e129e77e5a051b4c Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 19 Oct 2021 10:08:11 +0200
Subject: [PATCH] Learning rate grid search.
---
Scripts/downscale.sh | 14 ++++++++++++++
1 file changed, 14 insertions(+)
diff --git a/Scripts/downscale.sh b/Scripts/downscale.sh
index 233f571..a81a37a 100644
--- a/Scripts/downscale.sh
+++ b/Scripts/downscale.sh
@@ -14,6 +14,9 @@ LOSS=(L1Loss BernoulliGammaLoss MSELoss)
# weight decay values to test
LAMBDA=(0 0.000001 0.00001 0.0001 0.001 0.01 1)
+# learning rate values to test
+LR=(0.1 0.01 0.005 0.001 0.0005)
+
# iterate over loss functions
for loss in ${LOSS[@]}; do
@@ -35,6 +38,17 @@ for loss in ${LOSS[@]}; do
python climax/main/downscale_infer.py
done
+ # iterate over learning rate values
+ for lr in ${LR[@]}; do
+ # change weight regularization in configuration
+ sed -i "s/'lr':.*/'lr': $lr/" ./climax/main/config.py
+
+ # run downscaling
+ # python climax/main/downscale.py
+ python climax/main/downscale_train.py
+ python climax/main/downscale_infer.py
+ done
+
# for w in ${WET_DAY_THRESHOLDS[@]}; do
# change wet day threshold in configuration
# sed -i "s/min_amount\s*=.*/min_amount=$w)/" ./climax/main/config.py
--
GitLab