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(), |
