paddlespeech.t2s.models.vits.vits_updater 模块

class paddlespeech.t2s.models.vits.vits_updater.VITSEvaluator(model, criterions: Dict[str, Layer], dataloader: DataLoader, lambda_adv: float = 1.0, lambda_mel: float = 45.0, lambda_feat_match: float = 2.0, lambda_dur: float = 1.0, lambda_kl: float = 1.0, generator_first: bool = False, output_dir=None)

基础： StandardEvaluator

Attributes:

name

方法

__call__([trainer])	扩展的主要操作。
finalize(trainer)	训练完成时执行的操作。
initialize(trainer)	执行一次以获取正确的培训师状态的操作。
on_error(trainer, exc, tb)	处理训练过程中引发的错误，然后再进行最终处理。

评估
evaluate_core

evaluate_core(batch)

class paddlespeech.t2s.models.vits.vits_updater.VITSUpdater(model: Layer, optimizers: Dict[str, Optimizer], criterions: Dict[str, Layer], schedulers: Dict[str, LRScheduler], dataloader: DataLoader, generator_train_start_steps: int = 0, discriminator_train_start_steps: int = 100000, lambda_adv: float = 1.0, lambda_mel: float = 45.0, lambda_feat_match: float = 2.0, lambda_dur: float = 1.0, lambda_kl: float = 1.0, generator_first: bool = False, output_dir=None)

基础： StandardUpdater

Attributes:

updates_per_epoch

每个时代的更新者数量，由数据加载器的长度决定。

方法

new_epoch()	开始一个新的周期。
read_batch()	从数据加载器中读取一批，当数据耗尽时自动续订。
set_state_dict(state_dict)	为更新器设置状态字典。
state_dict()	一个Updater的状态字典，包括模型、优化器和更新器状态。
update_core(batch)	训练步骤的简单案例。

加载
保存
更新

update_core(batch)

训练步骤的简单案例。基本假设是： 单一模型； 单一优化器； 数据加载器中的一批数据只是模型的输入； 模型返回一个单一的损失，或者包含多个损失的字典。 每个批次参数更新，没有梯度累积。