From 1d75252371e9fb6eea63d8e1e8a0c36cbc084394 Mon Sep 17 00:00:00 2001
From: Patrick Esser <patrick.esser@gmx.net>
Date: Fri, 29 Jul 2022 10:41:44 +0000
Subject: [PATCH] reduce lr for ucg training

---
 scripts/slurm/v1_iahr_torch111_ucg/launcher.sh | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/slurm/v1_iahr_torch111_ucg/launcher.sh b/scripts/slurm/v1_iahr_torch111_ucg/launcher.sh
index d567991..d1268db 100755
--- a/scripts/slurm/v1_iahr_torch111_ucg/launcher.sh
+++ b/scripts/slurm/v1_iahr_torch111_ucg/launcher.sh
@@ -35,6 +35,9 @@ EXTRA="${EXTRA} data.params.min_size=512 data.params.max_pwatermark=0.4999"
 # unconditional guidance training
 EXTRA="${EXTRA} model.params.ucg_training.txt.p=0.1 model.params.ucg_training.txt.val=''"
 
+# reduce lr a bit
+EXTRA="${EXTRA} model.params.scheduler_config.params.f_max=[0.5]"
+
 # postfix
 EXTRA="${EXTRA} -f v1_iahr_torch111_ucg"