Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "93709d0363fbb714065609ea56ff99a42c85206e"
Commit 766e019a authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

just fix a typo in comments. 

PiperOrigin-RevId: 362151859
parent 0edeb7f6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
orbit/utils/tpu_summaries.py
View file @ 766e019a
...@@ -64,7 +64,7 @@ class OptionalSummariesFunction: ...@@ -64,7 +64,7 @@ class OptionalSummariesFunction:
return TrainFunctionWithSummaries(function, **kwargs) return TrainFunctionWithSummaries(function, **kwargs)
return functools.partial(TrainFunctionWithSummaries, **kwargs) return functools.partial(TrainFunctionWithSummaries, **kwargs)
The director can be applied directly to `train()` methods: The decorator can be applied directly to `train()` methods:
@train_function_with_summaries @train_function_with_summaries
def train(self, num_steps): def train(self, num_steps):
......
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