Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

remove old print

This commit is contained in:
Victor Hall 2023-05-11 19:17:19 -04:00
parent 9f2edb499d
commit 003b08992b
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
optimizer/optimizers.py
View File

@ -257,7 +257,6 @@ class EveryDreamOptimizer():
default_lr = 1e-6 default_lr = 1e-6
curr_lr = args.lr curr_lr = args.lr
text_encoder_lr_scale = 1.0
if local_optimizer_config is not None: if local_optimizer_config is not None:
betas = local_optimizer_config["betas"] or betas betas = local_optimizer_config["betas"] or betas
@ -269,8 +268,6 @@ class EveryDreamOptimizer():
curr_lr = args.lr curr_lr = args.lr
logging.info(f"Overriding LR from optimizer config with main config/cli LR setting: {curr_lr}") logging.info(f"Overriding LR from optimizer config with main config/cli LR setting: {curr_lr}")
print(f" * Using text encoder LR scale {text_encoder_lr_scale}")
if curr_lr is None: if curr_lr is None:
curr_lr = default_lr curr_lr = default_lr
logging.warning(f"No LR setting found, defaulting to {default_lr}") logging.warning(f"No LR setting found, defaulting to {default_lr}")