alu/tra-analysis
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-analysis.git synced 2025-04-03 01:33:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor analysis.py

Browse Source
This commit is contained in:
ltcptgeneral 2018-12-20 15:05:43 -06:00
parent 97a6e7c630
commit cb3b01863d
2 changed files with 1 additions and 2 deletions
data analysis
__pycache__
analysis.cpython-37.pyc
analysis.py

BIN
data analysis/__pycache__/analysis.cpython-37.pyc

Binary file not shown.

3
data analysis/analysis.py

@ -145,9 +145,8 @@ class error(ValueError):
def _init_device (setting, arg): #initiates computation device for ANNs
if setting == "cuda":
temp = setting + ":" + str(arg)
try:
return torch.device(temp if torch.cuda.is_available() else "cpu")
return torch.device(setting + ":" + str(arg) if torch.cuda.is_available() else "cpu")
except:
raise error("could not assign cuda or cpu")
elif setting == "cpu":