orbit.utils.EpochHelper

A helper class handle bookkeeping of epochs in custom training loops.

epoch_steps An integer indicating how many steps are in an epoch.
global_step A tf.Variable providing the current global step.

batch_index Index of the next batch within the current epoch.
current_epoch

Methods

epoch_begin

View source

Returns whether a new epoch should begin.

epoch_end

View source

Returns whether the current epoch should end.