From 871386bdaf100d7d21dcb725555e4c3f7d958992 Mon Sep 17 00:00:00 2001
From: janEbert <janpublicebert@posteo.net>
Date: Sun, 3 Nov 2024 15:53:20 +0100
Subject: [PATCH] Fix train step logging
---
pytorch-ddp-example/main.py | 2 +-
pytorch-fsdp-example/main.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/pytorch-ddp-example/main.py b/pytorch-ddp-example/main.py
index 7c23847..7b837e6 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 e1d16f1..55fba60 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}'
)
--
GitLab