diff --git a/pytorch-ddp-example/main.py b/pytorch-ddp-example/main.py index 7c238476f62b4c13d360ce03183d2b16be5e735f..7b837e6b9b298d2649dd73c791893581692af204 100644 --- a/pytorch-ddp-example/main.py +++ b/pytorch-ddp-example/main.py @@ -274,7 +274,7 @@ def main(): if step % log_step_interval == 0: print0( - f'[{epoch}/{epochs}; {step}] ' + f'[{epoch}/{epochs}; {step - 1}] ' f'loss: {loss:.5f}' ) diff --git a/pytorch-fsdp-example/main.py b/pytorch-fsdp-example/main.py index e1d16f1dd54b7464fe27448f51ad600e9ead7192..55fba60330f04d8a11b8e351931ec412481ef511 100644 --- a/pytorch-fsdp-example/main.py +++ b/pytorch-fsdp-example/main.py @@ -326,7 +326,7 @@ def main(): if step % log_step_interval == 0: print0( - f'[{epoch}/{epochs}; {step}] ' + f'[{epoch}/{epochs}; {step - 1}] ' f'loss: {loss:.5f}' )