2021-12-02 09:29:36 +08:00
|
|
|
|
.. py:class:: mindspore.train.callback.RunContext(original_args)
|
|
|
|
|
|
|
|
|
|
|
|
提供模型的相关信息。
|
|
|
|
|
|
|
|
|
|
|
|
在Model方法里提供模型的相关信息。
|
2021-12-28 20:07:42 +08:00
|
|
|
|
回调函数可以调用 `request_stop()` 方法来停止迭代。
|
2021-12-02 09:29:36 +08:00
|
|
|
|
|
|
|
|
|
|
**参数：**
|
|
|
|
|
|
|
2021-12-04 20:36:47 +08:00
|
|
|
|
- **original_args** (dict) - 模型的相关信息。
|
|
|
|
|
|
|
2021-12-02 09:29:36 +08:00
|
|
|
|
.. py:method:: get_stop_requested()
|
|
|
|
|
|
|
2021-12-28 20:07:42 +08:00
|
|
|
|
获取是否停止训练的标志。
|
2021-12-02 09:29:36 +08:00
|
|
|
|
|
|
|
|
|
|
**返回：**
|
2021-12-04 20:36:47 +08:00
|
|
|
|
|
2021-12-02 09:29:36 +08:00
|
|
|
|
bool，如果为True，则 `Model.train()` 停止迭代。
|
2021-12-04 20:36:47 +08:00
|
|
|
|
|
2021-12-02 09:29:36 +08:00
|
|
|
|
.. py:method:: original_args()
|
|
|
|
|
|
|
2021-12-28 20:07:42 +08:00
|
|
|
|
获取模型相关信息的对象。
|
2021-12-02 09:29:36 +08:00
|
|
|
|
|
|
|
|
|
|
**返回：**
|
2021-12-04 20:36:47 +08:00
|
|
|
|
|
2021-12-28 20:07:42 +08:00
|
|
|
|
dict，含有模型的相关信息的对象。
|
2021-12-04 20:36:47 +08:00
|
|
|
|
|
2021-12-02 09:29:36 +08:00
|
|
|
|
.. py:method:: request_stop()
|
|
|
|
|
|
|
|
|
|
|
|
在训练期间设置停止请求。
|
|
|
|
|
|
|
|
|
|
|
|
可以使用此函数请求停止训练。 `Model.train()` 会检查是否调用此函数。
|
