Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "3dead4d15fdb5aa60156e3530091cf963115e283"
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