Class TimeMonitor

Inheritance Relationships

Base Type

Class Documentation

class TimeMonitor : public mindspore::TrainCallBack

Public Functions

virtual void EpochBegin(const TrainCallBackData &cb_data) override

This method is called at the beginning of each epoch.

Parameters

cb_data[in] info about current execution

virtual CallbackRetValue EpochEnd(const TrainCallBackData &cb_data) override

This method is called after the run of each epoch.

Parameters

cb_data[in] info about current execution

Returns

indication if to continue in the train loop: RET_CONTINUE — continue training RET_STOP_TRAINING — stop training (e.g., due to achieved accuracy) RET_EXIT — Exit training (due to error of some sort)