tf.summary.experimental.set_step

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

tf.summary.experimental.set_step(
    step
)

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

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