该配置文件用于进行音频信号降噪任务的训练,主要包含以下部分:

  • '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 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录