"tools/vscode:/vscode.git/clone" did not exist on "e28768eafad82081c3c0e3773162865739caece3"
Commit 6ac8ca60 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Fix doc string after the code is updated.

PiperOrigin-RevId: 336339472
parent 571880ce
...@@ -74,8 +74,7 @@ class Trainer(orbit.StandardTrainer, orbit.StandardEvaluator): ...@@ -74,8 +74,7 @@ class Trainer(orbit.StandardTrainer, orbit.StandardEvaluator):
Args: Args:
config: An `ExperimentConfig` instance specifying experiment config. config: An `ExperimentConfig` instance specifying experiment config.
task: A base_task.Task instance. task: A base_task.Task instance.
model: tf.keras.Model instance. If provided, it will be used instead of model: The model instance, e.g. a tf.keras.Model instance.
building model using task.build_model(). Default to None.
optimizer: tf.optimizers.Optimizer instance. optimizer: tf.optimizers.Optimizer instance.
train: bool, whether or not this trainer will be used for training. train: bool, whether or not this trainer will be used for training.
default to True. default to True.
......
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