tf.compat.v2.summary.experimental.set_step

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

tf.compat.v2.summary.experimental.set_step(
    step
)

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

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