From f769d74b0f92acd1e1d35cd43e14717171e593da Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Mon, 2 Jan 2023 11:50:58 +0100 Subject: [PATCH] Fix typo in train_dreambooth_inpaint (#1885) Fix typo in train_dreambooth_inpaint. --- .../dreambooth_inpaint/train_dreambooth_inpaint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py index e1280ede..16cd0b71 100644 --- a/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py +++ b/examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py @@ -672,7 +672,7 @@ def main(): progress_bar = tqdm(range(global_step, args.max_train_steps), disable=not accelerator.is_local_main_process) progress_bar.set_description("Steps") - for epoch in range(first_epoch, args.num_epochs): + for epoch in range(first_epoch, args.num_train_epochs): unet.train() for step, batch in enumerate(train_dataloader): # Skip steps until we reach the resumed step