Есть ли замена для global_step в TensorFlow 2.0?

Question

Похоже, что global_step отсутствует в TensorFlow 2.0.

У меня есть несколько обратных вызовов, которые заинтересованы в текущем прогрессе обучения, и я не уверен, нужно ли мне использовать свой собственный счетчик шагов или вместо этого зависеть от количества эпох ...

Любые рекомендации по замене?

Answer 1

Сейчас лучше объявить наш собственный global_step = tf.Variable(1, name="global_step") и использовать его вручную.

Глядя на документацию, нет замены для tf.train.get_or_create_global_step и единственной части документации, котораяо step это экспериментальная часть модуля tf.summary: https://www.tensorflow.org/versions/r2.0/api_docs/python/tf/summary/experimental