espnet2.train.gan_trainer.GANTrainer

source

class espnet2.train.gan_trainer.GANTrainer

Bases: Trainer

Trainer for GAN-based training.

classmethod add_arguments(parser: ArgumentParser)

Add additional arguments for GAN-trainer.

classmethod build_options(args: Namespace) → TrainerOptions

Build options consumed by train(), eval(), and plot_attention().

classmethod train_one_epoch(model: Module, iterator: Iterable[Tuple[List[str], Dict[str, Tensor]]], optimizers: Sequence[Optimizer], schedulers: Sequence[AbsScheduler | None], scaler: GradScaler | None, reporter: SubReporter, summary_writer, options: GANTrainerOptions, distributed_option: DistributedOption) → bool

Train one epoch.

classmethod validate_one_epoch(model: Module, iterator: Iterable[Dict[str, Tensor]], reporter: SubReporter, options: GANTrainerOptions, distributed_option: DistributedOption) → None

Validate one epoch.