音频降噪模型训练配置 - TCNN 模型
该配置文件用于进行音频信号降噪任务的训练,主要包含以下部分:
- 'seed':随机数种子,用于复现实验结果。
- 'description':实验描述,可选。
- 'root_dir':实验根目录。
- 'cudnn_deterministic':是否开启CUDNN随机数生成器。
- 'trainer':训练器的参数设置。
- 'epochs':训练的轮数。
- 'save_checkpoint_interval':保存检查点的间隔。
- 'validation':验证器的参数设置。
- 'interval':验证的间隔。
- 'find_max':是否寻找最大的验证结果。
- 'custom':自定义的验证参数。
- 'visualize_audio_limit':可视化音频的数量限制。
- 'visualize_waveform_limit':可视化波形的数量限制。
- 'visualize_spectrogram_limit':可视化频谱图的数量限制。
- 'model':模型的参数设置。
- 'module':模型所在的模块名。
- 'main':模型的主函数名。
- 'args':模型的参数设置。
- 'loss_function':损失函数的参数设置。
- 'module':损失函数所在的模块名。
- 'main':损失函数的主函数名。
- 'args':损失函数的参数设置。
- 'optimizer':优化器的参数设置。
- 'lr':学习率。
- 'beta1':一阶矩估计的指数衰减率。
- 'beta2':二阶矩估计的指数衰减率。
- 'train_dataset':训练数据集的参数设置。
- 'module':训练数据集所在的模块名。
- 'main':训练数据集的主函数名。
- 'args':训练数据集的参数设置。
- 'noise_dataset':噪声数据集文件的路径。
- 'clean_dataset':干净数据集文件的路径。
- 'snr_list':噪声信噪比(SNR)的列表。
- 'limit':数据集的大小限制。
- 'offset':数据集的偏移量。
- 'mode':数据集的模式(训练/验证)。
- 'n_jobs':使用的CPU数量,-1表示使用所有可用的CPU。
- 'validation_dataset':验证数据集的参数设置。
- 'module':验证数据集所在的模块名。
- 'main':验证数据集的主函数名。
- 'args':验证数据集的参数设置。
- 'noise_dataset':噪声数据集文件的路径。
- 'clean_dataset':干净数据集文件的路径。
- 'snr_list':噪声信噪比(SNR)的列表。
- 'limit':数据集的大小限制。
- 'offset':数据集的偏移量。
- 'mode':数据集的模式(训练/验证)。
- 'n_jobs':使用的CPU数量,-1表示使用所有可用的CPU。
- 'train_dataloader':训练数据集加载器的参数设置。
- 'batch_size':批量大小。
- 'num_workers':使用的CPU数量。
- 'shuffle':是否打乱数据。
- 'pin_memory':是否将数据存储在CUDA固定内存中。
原文地址: https://www.cveoy.top/t/topic/nwGd 著作权归作者所有。请勿转载和采集!