paddlespeech.t2s.training.extension 模块

class paddlespeech.t2s.training.extension.Extension

基础： object

扩展以自定义Trainer的行为。

Attributes:

default_name

扩展的默认名称，默认的类名。

name

方法

__call__(trainer)	扩展的主要操作。
finalize(trainer)	训练完成时执行的动作。
initialize(trainer)	执行一次以获得正确的训练状态的操作。
on_error(trainer, exc, tb)	处理训练过程中在最终化之前引发的错误。

property default_name

扩展的默认名称，默认的类名。

finalize(trainer)

训练完成后执行的操作。 例如，可能需要关闭可视化工具。

initialize(trainer)

一次性执行的操作，以获取正确的训练器状态。它通常在训练之前被调用，但如果训练器恢复了带有快照扩展的状态，则也应该调用此方法。

name = None

on_error(trainer, exc, tb)

处理在最终确认之前训练期间引发的错误。

priority = 100

trigger = (1, 'iteration')

paddlespeech.t2s.training.extension.make_extension(trigger: Optional[Callable] = None, default_name: Optional[str] = None, priority: Optional[int] = None, finalizer: Optional[Callable] = None, initializer: Optional[Callable] = None, on_error: Optional[Callable] = None)

通过注入所需属性来创建一个类似扩展的对象。