EpochType.is_warmup()

EpochType.is_warmup()#

static EpochType.is_warmup(epoch_type)[source]#

Returns True if the epoch is part of the warmup phase. Implemented as a static method to make it jittable.

Return type:

bool