search

qitianwu / DIFFormer

The official implementation for ICLR23 spotlight paper "DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusion"
302 stars 32 forks source link

error #4

Closed ShuaiXuan closed 1 year ago

ShuaiXuan commented 1 year ago

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

qitianwu commented 1 year ago

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