iioSnail

/

ReaLiSe-for-csc

like 5

我下载了这个模型，cpu上运行正常，但是如果我想使用gpu时，将model和inputs都转为cuda，就会报错，提示“lengths参数需要一维的int64类型cpu参数，但是得到的是一维的long类型cuda参数”。
调试之后，发现在csc_model文件中ReaLiseForCSC类的forward方法中，调用torch.nn.utils.rnn.pack_padded_sequence方法之前，加上一个pho_lens = pho_lens.to("cpu")的转换，可以使代码在cpu和gpu上都能正常执行。

作者是否愿意对这个小问题进行一个修复呢？