tf.summary.experimental.set_step

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

tf.summary.experimental.set_step(
    step
)

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

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