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开始时设置开始时间。

参数:
epoch_end(run_context)[源代码]

在epoch结束时获得损失。

参数: