From d762d5992acf1a128de477d357e029648cf4ae6b Mon Sep 17 00:00:00 2001 From: Patrick Esser Date: Thu, 28 Jul 2022 21:55:17 +0000 Subject: [PATCH] take restored gs into account for decay --- scripts/slurm/v1_iahr_torch111/launcher.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/slurm/v1_iahr_torch111/launcher.sh b/scripts/slurm/v1_iahr_torch111/launcher.sh index 524f6b5..37a18c3 100755 --- a/scripts/slurm/v1_iahr_torch111/launcher.sh +++ b/scripts/slurm/v1_iahr_torch111/launcher.sh @@ -36,7 +36,7 @@ EXTRA="${EXTRA} data.params.min_size=512 data.params.max_pwatermark=0.4999" EXTRA="${EXTRA} -f v1_iahr_torch111" # time to decay -EXTRA="${EXTRA} model.params.scheduler_config.params.cycle_lengths=[50000] model.params.scheduler_config.params.f_min=[1e-6]" +EXTRA="${EXTRA} model.params.scheduler_config.params.cycle_lengths=[300000] model.params.scheduler_config.params.warm_up_steps=[250000] model.params.scheduler_config.params.f_min=[1e-6]" # custom logdir #EXTRA="${EXTRA} --logdir rlogs"