summaryrefslogtreecommitdiffstats
path: root/train_dreambooth.py
diff options
context:
space:
mode:
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r--train_dreambooth.py2
1 files changed, 1 insertions, 1 deletions
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():
907 on_before_optimize=on_before_optimize, 907 on_before_optimize=on_before_optimize,
908 on_after_optimize=on_after_optimize, 908 on_after_optimize=on_after_optimize,
909 ) 909 )
910 lr_finder.run(num_epochs=100, end_lr=1e3) 910 lr_finder.run(num_epochs=100, end_lr=1e2)
911 911
912 plt.savefig(basepath.joinpath("lr.png"), dpi=300) 912 plt.savefig(basepath.joinpath("lr.png"), dpi=300)
913 plt.close() 913 plt.close()