Skip to content

Commit 5c7a66d

Browse files
committed
[Enhance] Add cur_epoch property for trainer
1 parent 7826304 commit 5c7a66d

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

xtuner/v1/train/trainer.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -822,6 +822,15 @@ def cur_step(self) -> int:
822822
"""
823823
return self._cur_step
824824

825+
@property
826+
def cur_epoch(self) -> int:
827+
"""Get the current training epoch.
828+
829+
Returns:
830+
int | None: Current epoch number or None if not applicable.
831+
"""
832+
return self._cur_epoch
833+
825834
def _init_logger(self, log_dir: Path):
826835
# Logging system maybe need better design
827836
logger = get_logger()

0 commit comments

Comments
 (0)