Всем привет, как можно сохранять нейросеть tensorflow object_detection при тренировке

Question

Всем привет, как можно сохранять нейросеть tensorflow object_detection при тренировке

каждый пятидесятый шаг? Дело в том, что оно сохраняется спустя какой-то долгий срок на рандомном шагу, а к тому времени моя сетка уже переобучается.

Вот что я нашел на stackoverflow:
tf.keras.callbacks.ModelCheckpoint(
filepath, monitor='val_loss', verbose=0, save_best_only=False,
save_weights_only=False, mode='auto', save_freq=50, **kwargs)

Но проблема в том, что у меня в коде (я юзаю исходники) подобного кода вообще нету
Единственное что я нашел в своем коде связанное с save и checkpoint, это вот этот кусок кода:
flags.DEFINE_string('train_dir', '',
'Directory to save the checkpoints and training summaries.')