Closed wcqy-ye closed 6 months ago
为了保证训练完的模型可以支持torch.jit.script导出,在训练一开始会直接检查一遍模型
只需修改一下这个forward函数即可,参考ecapa_tdnn中的改法更改一下即可 https://github.com/wenet-e2e/wespeaker/blob/2968d940df84dfe70c2476b439bbe3113b7d3234/wespeaker/models/ecapa_tdnn.py#L66
那请问这个问题是为什么呢
现在判断是在script_model.save(os.path.join(model_dir, 'init.zip'))语句出现问题,但是感觉script_model = torch.jit.script(model)已经检查过了模型为什么还会出错呢 不太理解
问题已经解决了 是因为forward函数中有默认为None的参数
想请问一下为什么会出现这个问题 这就是一个很简单的res2conv1d
报错如下:
我在单机单卡训练的时候没有出现这个问题,因此不知道是不是多卡环境下导致对于变量的初始化等有要求,所以想请教一下