paddlespeech.t2s.models.melgan.multi_band_melgan_updater 模块

class paddlespeech.t2s.models.melgan.multi_band_melgan_updater.MBMelGANEvaluator(models: Dict[str, Layer], criterions: Dict[str, Layer], dataloader: DataLoader, lambda_aux: float = 1.0, lambda_adv: float = 1.0, output_dir: Optional[Path] = None)

基础： StandardEvaluator

Attributes:

name

方法

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

评估
evaluate_core

evaluate_core(batch)

class paddlespeech.t2s.models.melgan.multi_band_melgan_updater.MBMelGANUpdater(models: Dict[str, 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_aux: float = 1.0, lambda_adv: float = 1.0, output_dir: Optional[Path] = None)

基础： StandardUpdater

Attributes:

updates_per_epoch

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

方法

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

加载
保存
更新

update_core(batch)

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