Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
f48d3314
Unverified
Commit
f48d3314
authored
Mar 22, 2023
by
Tom Aarsen
Committed by
GitHub
Mar 22, 2023
Browse files
docs: Resolve incorrect type typo in trainer methods (#22316)
Resolve incorrect type typo in trainer methods
parent
0f68a7f4
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
src/transformers/trainer.py
src/transformers/trainer.py
+3
-3
No files found.
src/transformers/trainer.py
View file @
f48d3314
...
@@ -2915,7 +2915,7 @@ class Trainer:
...
@@ -2915,7 +2915,7 @@ class Trainer:
Pass a dataset if you wish to override `self.eval_dataset`. If it is a [`~datasets.Dataset`], columns
Pass a dataset if you wish to override `self.eval_dataset`. If it is a [`~datasets.Dataset`], columns
not accepted by the `model.forward()` method are automatically removed. It must implement the `__len__`
not accepted by the `model.forward()` method are automatically removed. It must implement the `__len__`
method.
method.
ignore_keys (`Lst[str]`, *optional*):
ignore_keys (`L
i
st[str]`, *optional*):
A list of keys in the output of your model (if it is a dictionary) that should be ignored when
A list of keys in the output of your model (if it is a dictionary) that should be ignored when
gathering predictions.
gathering predictions.
metric_key_prefix (`str`, *optional*, defaults to `"eval"`):
metric_key_prefix (`str`, *optional*, defaults to `"eval"`):
...
@@ -2980,7 +2980,7 @@ class Trainer:
...
@@ -2980,7 +2980,7 @@ class Trainer:
test_dataset (`Dataset`):
test_dataset (`Dataset`):
Dataset to run the predictions on. If it is an `datasets.Dataset`, columns not accepted by the
Dataset to run the predictions on. If it is an `datasets.Dataset`, columns not accepted by the
`model.forward()` method are automatically removed. Has to implement the method `__len__`
`model.forward()` method are automatically removed. Has to implement the method `__len__`
ignore_keys (`Lst[str]`, *optional*):
ignore_keys (`L
i
st[str]`, *optional*):
A list of keys in the output of your model (if it is a dictionary) that should be ignored when
A list of keys in the output of your model (if it is a dictionary) that should be ignored when
gathering predictions.
gathering predictions.
metric_key_prefix (`str`, *optional*, defaults to `"test"`):
metric_key_prefix (`str`, *optional*, defaults to `"test"`):
...
@@ -3314,7 +3314,7 @@ class Trainer:
...
@@ -3314,7 +3314,7 @@ class Trainer:
argument `labels`. Check your model's documentation for all accepted arguments.
argument `labels`. Check your model's documentation for all accepted arguments.
prediction_loss_only (`bool`):
prediction_loss_only (`bool`):
Whether or not to return the loss only.
Whether or not to return the loss only.
ignore_keys (`Lst[str]`, *optional*):
ignore_keys (`L
i
st[str]`, *optional*):
A list of keys in the output of your model (if it is a dictionary) that should be ignored when
A list of keys in the output of your model (if it is a dictionary) that should be ignored when
gathering predictions.
gathering predictions.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment