mindelec.solver.LossAndTimeMonitor
- class mindelec.solver.LossAndTimeMonitor(data_size, per_print_times=1)[源代码]
监控训练中的loss。
如果loss是NAN或INF,将终止训练。
说明
如果 per_print_times 为0,则不打印loss。
- 参数:
data_size (int) - 每个epoch数据集的批次数。
per_print_times (int) - 表示每隔多少个step打印一次loss。默认值:1。
- 异常:
ValueError - 如果 data_size 不是整数或小于零。
ValueError - 如果 per_print_times 不是整数或小于零。
- 支持平台:
Ascend
- epoch_begin(run_context)[源代码]
在epoch开始时设置开始时间。
- 参数:
run_context (RunContext) - 包含一些模型中的信息,详情请参考 mindspore.train.RunContext。
- epoch_end(run_context)[源代码]
在epoch结束时获得损失。
- 参数:
run_context (RunContext) - 包含一些模型中的信息,详情请参考 mindspore.train.RunContext。