Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
utils.py 112 Bytes
Newer Older
Quan (Andy) Gan's avatar
[Model][Feature] PinSage & Random Walk with Restart (#453)  
Quan (Andy) Gan committed
1
2
3
4
5
6
7 
import torch

def cuda(x):
    if torch.cuda.is_available():
        return x.cuda()
    else:
        return x