forked from huawei/mindspore2022
27 lines
877 B
ReStructuredText
27 lines
877 B
ReStructuredText
.. py:class:: mindspore.train.callback.LossMonitor(per_print_times=1, has_trained_epoch=0)
|
||
|
||
监控训练的loss。
|
||
|
||
如果loss是NAN或INF，则终止训练。
|
||
|
||
.. note::
|
||
如果 `per_print_times` 为0，则不打印loss。
|
||
|
||
**参数：**
|
||
|
||
- **per_print_times** (int) - 表示每隔多少个step打印一次loss。默认值：1。
|
||
- **has_trained_epoch** (int) - 表示已经训练了多少个epoch，如果设置了该参数，LossMonitor将监控该数值之后epoch的loss值。默认值：0。
|
||
|
||
**异常：**
|
||
|
||
- **ValueError** - 当 `per_print_times` 不是整数或小于零。
|
||
- **ValueError** - 当 `has_trained_epoch` 不是整数或小于零。
|
||
|
||
.. py:method:: step_end(run_context)
|
||
|
||
step结束时打印训练loss。
|
||
|
||
**参数：**
|
||
|
||
- **run_context** (RunContext) - 包含模型的相关信息。
|