Fix device inconsistency of model and data

CUAI / CorrectAndSmooth

[ICLR 2021] Combining Label Propagation and Simple Models Out-performs Graph Neural Networks (https://arxiv.org/abs/2010.13993)

285 stars 52 forks source link

Fix device inconsistency of model and data #8

Closed chewnglow closed 2 years ago

chewnglow commented 3 years ago

The code of sending data to device is data.to(device), while the model of that is model.cuda(). This may create device inconsistency when device is not set to the default device.