From be0a7d6ed6a4c939bafd8f3a35c16761854108d0 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 19 Oct 2021 16:13:27 +0200
Subject: [PATCH] Script to tune learning rate.

---
 Scripts/downscale.sh     |  6 ++++--
 Scripts/lr_range_test.sh | 24 ++++++++++++++++++++++++
 2 files changed, 28 insertions(+), 2 deletions(-)
 create mode 100644 Scripts/lr_range_test.sh

diff --git a/Scripts/downscale.sh b/Scripts/downscale.sh
index a81a37a..d607826 100644
--- a/Scripts/downscale.sh
+++ b/Scripts/downscale.sh
@@ -1,3 +1,5 @@
+#!/usr/bin/env bash
+
 # activate conda environment
 conda activate climax
 
@@ -22,9 +24,9 @@ for loss in ${LOSS[@]}; do
 
     # change loss function in configuration
     if [ "$loss" = "L1Loss" ] || [ "$loss" = "MSELoss" ]; then
-	sed -i "s/LOSS\s*=.*/LOSS=$loss()/" ./climax/main/config.py
+	    sed -i "s/LOSS\s*=.*/LOSS=$loss()/" ./climax/main/config.py
     else
-	sed -i "s/LOSS\s*=.*/LOSS=$loss(min_amount=1)/" ./climax/main/config.py
+	    sed -i "s/LOSS\s*=.*/LOSS=$loss(min_amount=1)/" ./climax/main/config.py
     fi
 
     # iterate over weight decay values
diff --git a/Scripts/lr_range_test.sh b/Scripts/lr_range_test.sh
new file mode 100644
index 0000000..1dc3c25
--- /dev/null
+++ b/Scripts/lr_range_test.sh
@@ -0,0 +1,24 @@
+#!/usr/bin/env bash
+
+# activate conda environment
+conda activate climax
+
+# move to project repository
+cd ~/git/climax
+
+# loss functions
+LOSS=(L1Loss BernoulliGammaLoss BernoulliWeibullLoss MSELoss)
+
+# iterate over loss functions
+for loss in ${LOSS[@]}; do
+
+    # change loss function in configuration
+    if [ "$loss" = "L1Loss" ] || [ "$loss" = "MSELoss" ]; then
+	    sed -i "s/LOSS\s*=.*/LOSS=$loss()/" ./climax/main/config.py
+    else
+	    sed -i "s/LOSS\s*=.*/LOSS=$loss(min_amount=1)/" ./climax/main/config.py
+    fi
+
+    # run learning rate range test
+    python climax/main/lr_range_test.py
+done
-- 
GitLab