paddlespeech.vector.exps.ge2e.train模块

class paddlespeech.vector.exps.ge2e.train.Ge2eExperiment(config, args)

基础: ExperimentBase

Attributes:

parallel

一个标志，指示实验是否应该使用多进程运行。

方法

close()	关闭可视化工具以避免训练后卡住
dump_config()	保存此实验使用的配置。
init_parallel()	初始化多进程训练的环境。
new_epoch()	重置训练加载器并增加epoch。
read_batch()	从train_loader读取一批数据。
resume_or_load()	从输出目录中的最新检查点恢复或加载指定的检查点。
run()	实验设置后的常规流程。
save()	保存检查点（模型参数和优化器状态）。
setup()	设置实验。
setup_checkpointer()	创建一个用于保存检查点的目录。
setup_dataloader()	设置训练数据加载器和验证数据加载器。
setup_logger()	初始化一个文本日志记录器以记录实验。
setup_model()	设置模型、标准和优化器等。
setup_output_dir()	创建一个用于输出的目录。
setup_visualizer()	初始化一个可视化工具以记录实验。
train()	训练过程。
train_batch()	训练循环。
valid()	验证。

setup_dataloader()

设置训练数据加载器和验证数据加载器。子类应实现此方法。

setup_model()

设置模型、标准和优化器等。子类应该实现此方法。

train_batch()

训练循环。子类应实现此方法。

valid()

验证。子类应该实现此方法。

paddlespeech.vector.exps.ge2e.train.main(config, args)

paddlespeech.vector.exps.ge2e.train.main_sp(config, args)