orbit.utils.create_global_step | TensorFlow v2.16.1 (original) (raw)
orbit.utils.create_global_step
Stay organized with collections Save and categorize content based on your preferences.
Creates a tf.Variable suitable for use as a global step counter.
orbit.utils.create_global_step() -> tf.Variable
Creating and managing a global step variable may be necessary forAbstractTrainer subclasses that perform multiple parameter updates perController "step", or use different optimizers on different steps.
In these cases, an optimizer.iterations property generally can't be used directly, since it would correspond to parameter updates instead of iterations in the Controller's training loop. Such use cases should simply callstep.assign_add(1) at the end of each step.
| Returns |
|---|
| A non-trainable scalar tf.Variable of dtype tf.int64, with only the first replica's value retained when synchronizing across replicas in a distributed setting. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2025-04-18 UTC.