search

fjiang9 / NKF-AEC

Acoustic Echo Cancellation with Nerual Kalman Filtering
239 stars 60 forks source link

关于损失函数计算参数设置的问题 #18

Open Hzyyyy opened 10 months ago

Hzyyyy commented 10 months ago

您好，在复现模型训练的过程中，遇到了损失函数值异常的问题，有以下几个问题想请教一下作者：

  1. 训练数据直接从AEC-challenge官方下载（我用的链接是https://github.com/microsoft/AEC-Challenge/tree/main/datasets/synthetic_fullband ）完成之后，我进行了将采样频率从48khz降到16khz操作，同时模型在读取训练数据(近端为mic.wav，远端为echo.wav，真值为target.wav)的时候会随机选择1s的片段（后续因为nan的出现改成了随机选择0.5s），除此之外作者还有对数据进行其他的操作吗？您是否也是用这样的方式获取到的训练数据？如果不是的话想问一下您是从哪里下载得到的数据？
  2. 请问作者是用什么样的方式对神经网络的参数进行初始化的？在我复现的过程中发现如果不是用全0进行weights和bias参数初始化的话损失函数计算会出现nan的情况，您是否也遇到过这样的情况？请问是怎么解决的？
  3. 论文里提到损失函数对echo进行L2损失函数计算，是否可以直接调用nn.MSELoss()对回声进行计算？ 期待您的解答。