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