Skip to content

Commit 03f74e7

Browse files
committed
Use self.args
1 parent 1c33be0 commit 03f74e7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/transformers/trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1837,7 +1837,7 @@ def _inner_training_loop(
18371837
step = -1
18381838
for step, inputs in enumerate(epoch_iterator):
18391839
total_batched_samples += 1
1840-
1840+
18411841
if self.args.include_num_input_tokens_seen:
18421842
main_input_name = getattr(self.model, "main_input_name", "input_ids")
18431843
if main_input_name not in inputs:
@@ -2650,7 +2650,7 @@ def log(self, logs: Dict[str, float]) -> None:
26502650
"""
26512651
if self.state.epoch is not None:
26522652
logs["epoch"] = round(self.state.epoch, 2)
2653-
if args.include_num_input_tokens_seen:
2653+
if self.args.include_num_input_tokens_seen:
26542654
logs["num_input_tokens_seen"] = self.state.num_input_tokens_seen
26552655

26562656
output = {**logs, **{"step": self.state.global_step}}

0 commit comments

Comments
 (0)