alu/tra-analysis
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-analysis.git synced 2024-12-26 17:49:09 +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 c5d087dada
commit dc6f896071
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data analysis/cudaregress.py
View File

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