Set device bc I apparently forgot to do that

2025-09-06 15:07:21 +00:00 · 2019-09-23 00:01:31 -05:00
parent c5d087dada
commit dc6f896071
1 changed files with 2 additions and 0 deletions
--- a/analysis/cudaregress.py
+++ b/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