Learning rate grid search.

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