2022-09-15 23:44:53 +08:00
|
|
|
|
mindspore.train.TimeMonitor
|
|
|
|
|
|
===========================
|
2022-05-10 11:59:45 +08:00
|
|
|
|
|
2022-09-15 23:44:53 +08:00
|
|
|
|
.. py:class:: mindspore.train.TimeMonitor(data_size=None)
|
2021-12-02 09:29:36 +08:00
|
|
|
|
|
2022-05-18 11:11:03 +08:00
|
|
|
|
监控训练或推理的时间。
|
2021-12-02 09:29:36 +08:00
|
|
|
|
|
2022-07-22 10:22:12 +08:00
|
|
|
|
参数：
|
|
|
|
|
|
- **data_size** (int) - 表示每隔多少个step打印一次信息。如果程序在训练期间获取到Model的 `batch_num` ，则将把 `data_size` 设为 `batch_num` ，否则将使用 `data_size` 。默认值：None。
|
2021-12-04 20:36:47 +08:00
|
|
|
|
|
2022-07-22 10:22:12 +08:00
|
|
|
|
异常：
|
|
|
|
|
|
- **ValueError** - `data_size` 不是正整数。
|
2022-01-13 16:18:53 +08:00
|
|
|
|
|
2021-12-02 09:29:36 +08:00
|
|
|
|
.. py:method:: epoch_begin(run_context)
|
|
|
|
|
|
|
|
|
|
|
|
在epoch开始时记录时间。
|
|
|
|
|
|
|
2022-07-22 10:22:12 +08:00
|
|
|
|
参数：
|
2022-09-27 16:28:21 +08:00
|
|
|
|
- **run_context** (RunContext) - 包含模型的相关信息。详情请参考 :class:`mindspore.train.RunContext`。
|
2021-12-04 20:36:47 +08:00
|
|
|
|
|
2021-12-02 09:29:36 +08:00
|
|
|
|
.. py:method:: epoch_end(run_context)
|
|
|
|
|
|
|
|
|
|
|
|
在epoch结束时打印epoch的耗时。
|
|
|
|
|
|
|
2022-07-22 10:22:12 +08:00
|
|
|
|
参数：
|
2022-09-27 16:28:21 +08:00
|
|
|
|
- **run_context** (RunContext) - 包含模型的相关信息。详情请参考 :class:`mindspore.train.RunContext`。
|
