Skip to content

Commit cc84004

Browse files
FelixAbrahamssonsamedii
authored andcommitted
improve: epochs from 1 -> max_epochs + 1
1 parent 66a546b commit cc84004

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lantern/epochs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
def Epochs(max_epochs):
22
"""Simple generator that prints the current epoch"""
3-
for epoch in range(max_epochs):
4-
print(f"------ epoch: {epoch + 1} / {max_epochs} ------")
3+
for epoch in range(1, max_epochs + 1):
4+
print(f"------ epoch: {epoch} / {max_epochs} ------")
55
yield epoch

0 commit comments

Comments
 (0)