Skip to content

文本分类样例代码GPU运行报错 #349

Open
@503718696

Description

@503718696

文本分类样例代码在CPU模式下正常运行,但是GPU运行报错,错误如下:
RuntimeError: 'lengths' argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions