modelscope-funasr这个问题怎么解决?

modelscope-funasr这个问题怎么解决?Expected tensor for argument #1 'indices' to have one of the following scalar types: Long, Int; but got torch.cuda.DoubleTensor instead (while checking arguments for embedding)
https://github.com/alibaba-damo-academy/FunASR/issues/1386
打扰一下,我看到了这个issue,pull了新的代码pip install了,还是有这个问题。
torch:1.10.1
funasr:1.11

展开
收起
三分钟热度的鱼 2024-03-06 20:33:36 132 分享 版权
1 条回答
写回答
取消 提交回答
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    这个问题可能是由于在模型中使用了错误的数据类型导致的。根据错误信息,期望的张量类型是Long或Int,但是得到的是torch.cuda.DoubleTensor类型。

    要解决这个问题,你可以尝试将输入数据的类型转换为Long或Int类型。你可以使用PyTorch中的to()方法或者type()方法来实现这个转换。例如:

    input_tensor = input_tensor.to(torch.long)
    

    或者

    input_tensor = input_tensor.type(torch.LongTensor)
    

    请确保在进行类型转换之前,你已经将输入数据移动到了GPU上(如果需要的话)。希望这可以帮助你解决问题!

    2024-03-08 19:46:37
    赞同 展开评论

包含命名实体识别、文本分类、分词、关系抽取、问答、推理、文本摘要、情感分析、机器翻译等多个领域

热门讨论

热门文章

还有其他疑问?
咨询AI助理