Unverified Commit 8d59385f authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix TrainingArguments documentation (#19162)

* Fix TrainingArguments documentation

* Fix TFTrainingArguments documentation
parent 3a396c59
...@@ -119,7 +119,6 @@ class OptimizerNames(ExplicitEnum): ...@@ -119,7 +119,6 @@ class OptimizerNames(ExplicitEnum):
@dataclass @dataclass
class TrainingArguments: class TrainingArguments:
framework = "pt"
""" """
TrainingArguments is the subset of the arguments we use in our example scripts **which relate to the training loop TrainingArguments is the subset of the arguments we use in our example scripts **which relate to the training loop
itself**. itself**.
...@@ -500,6 +499,7 @@ class TrainingArguments: ...@@ -500,6 +499,7 @@ class TrainingArguments:
Whether to use Apple Silicon chip based `mps` device. Whether to use Apple Silicon chip based `mps` device.
""" """
framework = "pt"
output_dir: str = field( output_dir: str = field(
metadata={"help": "The output directory where the model predictions and checkpoints will be written."}, metadata={"help": "The output directory where the model predictions and checkpoints will be written."},
) )
......
...@@ -28,7 +28,6 @@ if is_tf_available(): ...@@ -28,7 +28,6 @@ if is_tf_available():
@dataclass @dataclass
class TFTrainingArguments(TrainingArguments): class TFTrainingArguments(TrainingArguments):
framework = "tf"
""" """
TrainingArguments is the subset of the arguments we use in our example scripts **which relate to the training loop TrainingArguments is the subset of the arguments we use in our example scripts **which relate to the training loop
itself**. itself**.
...@@ -162,6 +161,7 @@ class TFTrainingArguments(TrainingArguments): ...@@ -162,6 +161,7 @@ class TFTrainingArguments(TrainingArguments):
Whether to activate the XLA compilation or not. Whether to activate the XLA compilation or not.
""" """
framework = "tf"
tpu_name: Optional[str] = field( tpu_name: Optional[str] = field(
default=None, default=None,
metadata={"help": "Name of TPU"}, metadata={"help": "Name of TPU"},
......
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