Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "6d8834f08f7b5243cba925665f4a5a7296142383"
Unverified Commit 4781afd0 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Clarify arg class (#5916)

parent 8e0bcb56
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/trainer.py
View file @ 4781afd0
...@@ -144,9 +144,9 @@ class Trainer: ...@@ -144,9 +144,9 @@ class Trainer:
data_collator (:obj:`DataCollator`, `optional`, defaults to :func:`~transformers.default_data_collator`): data_collator (:obj:`DataCollator`, `optional`, defaults to :func:`~transformers.default_data_collator`):
The function to use to from a batch from a list of elements of :obj:`train_dataset` or The function to use to from a batch from a list of elements of :obj:`train_dataset` or
:obj:`eval_dataset`. :obj:`eval_dataset`.
train_dataset (:obj:`Dataset`, `optional`): train_dataset (:obj:`torch.utils.data.dataset.Dataset`, `optional`):
The dataset to use for training. The dataset to use for training.
eval_dataset (:obj:`Dataset`, `optional`): eval_dataset (:obj:`torch.utils.data.dataset.Dataset`, `optional`):
The dataset to use for evaluation. The dataset to use for evaluation.
compute_metrics (:obj:`Callable[[EvalPrediction], Dict]`, `optional`): compute_metrics (:obj:`Callable[[EvalPrediction], Dict]`, `optional`):
The function that will be used to compute metrics at evaluation. Must take a The function that will be used to compute metrics at evaluation. Must take a
......
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