quaterion.train.callbacks.metrics_callback 模块

class MetricsCallback

基类: Callback

on_sanity_check_end(trainer: Trainer, trainable_model: LightningModule) → None

在验证完整性检查结束时调用。

on_train_epoch_end(trainer: Trainer, trainable_model: LightningModule) → None

在训练 epoch 结束时调用。

要在 epoch 结束时访问所有批次输出，可以通过以下任一方式：

1. 在 LightningModule 中实现 training_epoch_end 并通过模块访问输出，或者

2. 在回调实现内部跨训练批次钩子缓存数据，以便在此钩子中进行后处理。

on_train_epoch_start(trainer: Trainer, trainable_model: LightningModule) → None

在训练 epoch 开始时调用。

on_validation_epoch_end(trainer: Trainer, trainable_model: LightningModule) → None

在验证 epoch 结束时调用。

on_validation_epoch_start(trainer: Trainer, trainable_model: LightningModule) → None

在验证 epoch 开始时调用。

static reset_metrics(trainable_model)