liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-15 08:46:56 +08:00
Code Activity

Add return for use_gpu..

Browse Source
This commit is contained in:
D-X-Y
2021-03-11 19:28:00 -08:00
parent 593553f573
commit 1d435248e2
11 changed files with 31 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_gats_ts.py
View File

@@ -172,7 +172,7 @@ class GATs(Model):
@property
def use_gpu(self):
self.device != torch.device("cpu")
return self.device != torch.device("cpu")
def mse(self, pred, label):
loss = (pred - label) ** 2