tf.summary.experimental.set_step

Устанавливает значение шага для записи сводок по умолчанию в текущей потоке.

tf.summary.experimental.set_step(
    step
)

Для удобства, эта функция устанавливает значение параметра step, используемого в функциях записи сводок в других частях API, чтобы его не нужно было явно передавать в каждом вызове. Значение может быть константой или переменной и может быть получено с помощью tf.summary.experimental.get_step().

Примечание: при использовании с @tf.functions значение шага будет захвачено в момент трассировки функции, поэтому изменения шага вне функции не будут отражены внутри функции, если не используется tf.Variable шага.