search

wenet-e2e / wespeaker

Research and Production Oriented Speaker Verification, Recognition and Diarization Toolkit
Apache License 2.0
630 stars 109 forks source link

想请教训练中的问题 #275

Closed wcqy-ye closed 6 months ago

wcqy-ye commented 6 months ago

想请问一下为什么会出现这个问题 这就是一个很简单的res2conv1d image 报错如下： image 我在单机单卡训练的时候没有出现这个问题，因此不知道是不是多卡环境下导致对于变量的初始化等有要求，所以想请教一下

JiJiJiang commented 6 months ago

为了保证训练完的模型可以支持torch.jit.script导出，在训练一开始会直接检查一遍模型 image

只需修改一下这个forward函数即可，参考ecapa_tdnn中的改法更改一下即可 https://github.com/wenet-e2e/wespeaker/blob/2968d940df84dfe70c2476b439bbe3113b7d3234/wespeaker/models/ecapa_tdnn.py#L66

wcqy-ye commented 6 months ago

那请问这个问题是为什么呢 image 现在判断是在script_model.save(os.path.join(model_dir, 'init.zip'))语句出现问题，但是感觉script_model = torch.jit.script(model)已经检查过了模型为什么还会出错呢 不太理解 image

wcqy-ye commented 6 months ago

问题已经解决了 是因为forward函数中有默认为None的参数