Class TimeMonitor
Defined in File time_monitor.h
Inheritance Relationships
Base Type
public mindspore::TrainCallBack
(Class TrainCallBack)
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)
-
virtual void EpochBegin(const TrainCallBackData &cb_data) override