From 6df1fc46daca9c289f1d7f7524e01deac5c92fd1 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Tue, 27 Dec 2022 13:58:48 +0100 Subject: Improved learning rate finder --- train_dreambooth.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'train_dreambooth.py') diff --git a/train_dreambooth.py b/train_dreambooth.py index a62cec9..325fe90 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -970,9 +970,8 @@ def main(): avg_loss_val.update(loss.detach_(), bsz) avg_acc_val.update(acc.detach_(), bsz) - if accelerator.sync_gradients: - local_progress_bar.update(1) - global_progress_bar.update(1) + local_progress_bar.update(1) + global_progress_bar.update(1) logs = { "val/loss": avg_loss_val.avg.item(), -- cgit v1.2.3-54-g00ecf