oa::CbLrScheduler

oa::LrSchedulerCallback Applies any oa::LRScheduler to the optimizer at each step. Use oa::CosineScheduler, oa::OneCycleScheduler, oa::WarmupScheduler, etc. from <oa /ml/optim.h>.

Inheritance

public CbTraining · public Callback

Public Methods

void oa::CbLrScheduler::onStepEnd(oa::ItTraining & inIter)

Constructor & Destructor Documentation

oa::CbLrScheduler::CbLrScheduler( oa::LRScheduler & inScheduler, oa::Optimizer & inOpt )
No public source comment is attached to this declaration.

Parameters

inScheduler
oa::LRScheduler &

inOpt
oa::Optimizer &

Public Method Documentation

void oa::CbLrScheduler::onStepEnd( oa::ItTraining & inIter )
No public source comment is attached to this declaration.

Parameters

inIter
oa::ItTraining &

Returns

void

The declared return value.