From c7ffd09fb4432a8ce4b64d56eecdc932a475adeb Mon Sep 17 00:00:00 2001 From: "Michael Pilosov, PhD" Date: Sun, 28 Jan 2024 02:44:36 +0000 Subject: [PATCH] lr scheduler --- newsearch.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/newsearch.py b/newsearch.py index b632f6f..93990a9 100644 --- a/newsearch.py +++ b/newsearch.py @@ -90,13 +90,13 @@ python newmain.py fit \ --trainer.callbacks.init_args.save_interval 0 \ --optimizer torch.optim.{opt} \ --optimizer.init_args.lr {lr} \ ---trainer.callbacks+ lightning.pytorch.callbacks.LearningRateFinder -# --lr_scheduler lightning.pytorch.cli.ReduceLROnPlateau \ -# --lr_scheduler.init_args.monitor hp_metric \ -# --lr_scheduler.init_args.factor 0.05 \ -# --lr_scheduler.init_args.patience 5 \ -# --lr_scheduler.init_args.cooldown 10 \ -# --lr_scheduler.init_args.verbose true +--trainer.callbacks+ lightning.pytorch.callbacks.LearningRateFinder \ +--lr_scheduler lightning.pytorch.cli.ReduceLROnPlateau \ +--lr_scheduler.init_args.monitor hp_metric \ +--lr_scheduler.init_args.factor 0.05 \ +--lr_scheduler.init_args.patience 5 \ +--lr_scheduler.init_args.cooldown 10 \ +--lr_scheduler.init_args.verbose true """ # job_name = f"color2_{bs}_{a}_{lr:2.2e}" # job_plugin.run(cmd, machine=Machine.T4, name=job_name)