AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[train_unconditional] fix LR scheduler init (#2010)

Browse Source 
fix lr scheduler
This commit is contained in:
Suraj Patil
2023-01-17 10:11:46 +01:00
committed by GitHub
parent b2ea8a84e9
commit f861cde14f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/unconditional_image_generation/train_unconditional.py
View File

@@ -338,8 +338,8 @@ def main(args):
lr_scheduler = get_scheduler(
args.lr_scheduler,
optimizer=optimizer,
num_warmup_steps=args.lr_warmup_steps,
num_training_steps=(len(train_dataloader) * args.num_epochs) // args.gradient_accumulation_steps,
num_warmup_steps=args.lr_warmup_steps * args.gradient_accumulation_steps,
num_training_steps=(len(train_dataloader) * args.num_epochs),
)
model, optimizer, train_dataloader, lr_scheduler = accelerator.prepare(