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) - 训练运行的上下文。

epoch_end(run_context)

在epoch结束时获得损失。

参数：

• run_context (RunContext) - 训练运行的上下文。