qitianwu / DIFFormer

The official implementation for ICLR23 spotlight paper "DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusion"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

error

ShuaiXuan opened this issue · comments

commented

RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

Hi, thank you for running our codes! Can you provide more details about what tasks/scripts you run and where the error is reported?

The current codes can be run smoothly when tested on our side