Skip to content

节点特征 #433

Description

@Flians

❓ Questions & Help

你好!请问我自定义的图数据集里面,节点特征为categorical,然后我使用了one-hot编码,使用experiment(model="gin", dataset=dataset)训练时,显示如下错误:
cogdl/operators/spmm/spmm.cpp:39: at::Tensor csr_spmm(at::Tensor, at::Tensor, at::Tensor, at::Tensor): Assertion B.dtype() == torch::kFloat32 || B.dtype() == torch::kFloat16' failed.`

请问有什么解决方法吗?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions