28 lines
2.1 KiB
ReStructuredText
28 lines
2.1 KiB
ReStructuredText
mindspore.train.LambdaCallback
|
||
==============================
|
||
|
||
.. py:class:: mindspore.train.LambdaCallback(on_train_epoch_begin=None, on_train_epoch_end=None, on_train_step_begin=None, on_train_step_end=None, on_train_begin=None, on_train_end=None, on_eval_epoch_begin=None, on_eval_epoch_end=None, on_eval_step_begin=None, on_eval_step_end=None, on_eval_begin=None, on_eval_end=None)
|
||
|
||
用于自定义简单的callback。
|
||
|
||
使用匿名函数构建callback，定义的匿名函数将在 `mindspore.train.Model.{train | eval | fit}` 的对应阶段被调用。
|
||
|
||
请注意，callback的每个阶段都需要一个位置参数：`run_context`。
|
||
|
||
.. warning::
|
||
这是一个实验性API，后续可能修改或删除。
|
||
|
||
参数：
|
||
- **on_train_epoch_begin** (Function，可选) - 训练的每个epoch开始时被调用。默认值： ``None`` 。
|
||
- **on_train_epoch_end** (Function，可选) - 训练的每个epoch结束时被调用。默认值： ``None`` 。
|
||
- **on_train_step_begin** (Function，可选) - 训练的每个step开始时被调用。默认值： ``None`` 。
|
||
- **on_train_step_end** (Function，可选) - 训练的每个step结束时被调用。默认值： ``None`` 。
|
||
- **on_train_begin** (Function，可选) - 模型训练开始时被调用。默认值： ``None`` 。
|
||
- **on_train_end** (Function，可选) - 模型训练结束时被调用。默认值： ``None`` 。
|
||
- **on_eval_epoch_begin** (Function，可选) - 推理的每个epoch开始时被调用。默认值： ``None`` 。
|
||
- **on_eval_epoch_end** (Function，可选) - 推理的每个epoch结束时被调用。默认值： ``None`` 。
|
||
- **on_eval_step_begin** (Function，可选) - 推理的每个step开始时被调用。默认值： ``None`` 。
|
||
- **on_eval_step_end** (Function，可选) - 推理的每个step结束时被调用。默认值： ``None`` 。
|
||
- **on_eval_begin** (Function，可选) - 模型推理开始时被调用。默认值： ``None`` 。
|
||
- **on_eval_end** (Function，可选) - 模型推理结束时被调用。默认值： ``None`` 。
|