alu/tra-analysis
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-analysis.git synced 2025-04-26 19:37:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set device bc I apparently forgot to do that

Browse Source
This commit is contained in:
jlevine18 2019-09-23 00:01:31 -05:00 committed by GitHub
parent d76075fbfa
commit 4522c9561b
1 changed files with 2 additions and 0 deletions

2
data analysis/cudaregress.py

@ -38,6 +38,8 @@ __all__ = [
import torch
#set device
device='cuda:0' if torch.cuda.is_available() else 'cpu'
#todo: document completely