Program Listing for File time_monitor.h

↰ Return to documentation for file (include/runtime/include/api/callback/time_monitor.h)

#ifndef MINDSPORE_INCLUDE_API_CALLBACK_TIME_MONITOR_H
#define MINDSPORE_INCLUDE_API_CALLBACK_TIME_MONITOR_H

#include <cstddef>
#include <string>
#include <vector>
#include <memory>
#include "include/api/callback/callback.h"

namespace mindspore {

class MS_API TimeMonitor : public TrainCallBack {
 public:
  virtual ~TimeMonitor() = default;
  void EpochBegin(const TrainCallBackData &cb_data) override;
  CallbackRetValue EpochEnd(const TrainCallBackData &cb_data) override;
};
}  // namespace mindspore
#endif  // MINDSPORE_INCLUDE_API_CALLBACK_TIME_MONITOR_H