Program Listing for File loss_monitor.h
↰ Return to documentation for file (include/runtime/include/api/callback/loss_monitor.h
)
#ifndef MINDSPORE_INCLUDE_API_CALLBACK_LOSS_MONITOR_H
#define MINDSPORE_INCLUDE_API_CALLBACK_LOSS_MONITOR_H
#include <cstddef>
#include <vector>
#include <utility>
#include "include/api/callback/callback.h"
namespace mindspore {
class MS_API LossMonitor : public TrainCallBack {
public:
explicit LossMonitor(int print_every_n_steps = INT_MAX);
virtual ~LossMonitor();
const std::vector<GraphPoint> &GetLossPoints();
};
} // namespace mindspore
#endif // MINDSPORE_INCLUDE_API_CALLBACK_LOSS_MONITOR_H