Skip to content

GitLab

Commit 3327c829 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

just fix a typo in comments. 

PiperOrigin-RevId: 362151859
parent 5b952c08
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
orbit/utils/tpu_summaries.py
View file @ 3327c829
...@@ -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