From 07e8111a98afbf912ba99a28e2d7ea647b2d29fc Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 13 Jan 2023 09:17:44 +0100 Subject: Added TI decay start offset --- train_dreambooth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_dreambooth.py') diff --git a/train_dreambooth.py b/train_dreambooth.py index da3a075..0fe590f 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -907,7 +907,7 @@ def main(): on_before_optimize=on_before_optimize, on_after_optimize=on_after_optimize, ) - lr_finder.run(num_epochs=100, end_lr=1e3) + lr_finder.run(num_epochs=100, end_lr=1e2) plt.savefig(basepath.joinpath("lr.png"), dpi=300) plt.close() -- cgit v1.2.3-54-g00ecf