Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "93a756266ce39ca3ec884750152b43d91a59ab27"
Switch branch/tag
Find file Normal viewHistoryPermalink
configs.py 121 Bytes
Newer Older
Xiang Gao's avatar
torchani 0.1  
Xiang Gao committed
1
2
3 
import torch
data_path = 'data/ANI-1x_complete.h5'
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")