How to modify the code to make the model run on the gpu？ - Githubissues

YGZWQZD / LAMDA-SSL

30 Semi-Supervised Learning Algorithms

MIT License

184 stars 16 forks source link

How to modify the code to make the model run on the gpu？ #11

Open Dawn-sir opened 1 year ago

quanguanlangze commented 8 months ago

device=‘cpu’ to device=‘cuda:0’