Skip to content

GitLab

Unverified Commit 3ed5e97b authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix Trainer with remove_unused_columns=False (#11382) 

* Fix Trainer with remove_unused_columns=False

* Typo
parent 0f3ad150
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/trainer.py
View file @ 3ed5e97b
...@@ -482,7 +482,7 @@ class Trainer: ...@@ -482,7 +482,7 @@ class Trainer:
def _remove_unused_columns(self, dataset: "datasets.Dataset", description: Optional[str] = None): def _remove_unused_columns(self, dataset: "datasets.Dataset", description: Optional[str] = None):
if not self.args.remove_unused_columns: if not self.args.remove_unused_columns:
return return dataset
if self._signature_columns is None: if self._signature_columns is None:
# Inspect model forward signature to keep only the arguments it accepts. # Inspect model forward signature to keep only the arguments it accepts.
signature = inspect.signature(self.model.forward) signature = inspect.signature(self.model.forward)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment