diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 5 |
1 files changed, 2 insertions, 3 deletions
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(): | |||
970 | avg_loss_val.update(loss.detach_(), bsz) | 970 | avg_loss_val.update(loss.detach_(), bsz) |
971 | avg_acc_val.update(acc.detach_(), bsz) | 971 | avg_acc_val.update(acc.detach_(), bsz) |
972 | 972 | ||
973 | if accelerator.sync_gradients: | 973 | local_progress_bar.update(1) |
974 | local_progress_bar.update(1) | 974 | global_progress_bar.update(1) |
975 | global_progress_bar.update(1) | ||
976 | 975 | ||
977 | logs = { | 976 | logs = { |
978 | "val/loss": avg_loss_val.avg.item(), | 977 | "val/loss": avg_loss_val.avg.item(), |