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

Fix bugs in use_gpu

Browse Source
This commit is contained in:
D-X-Y
2021-03-11 19:10:32 -08:00
parent db59713d36
commit d38b8d6001
9 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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