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

update

Browse Source
This commit is contained in:
Alex Wang
2020-11-25 14:27:19 +08:00
parent 6b90c6d066
commit fcbafde741
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
qlib/contrib/model/pytorch_sfm.py
View File

@@ -420,11 +420,11 @@ class SFM(Model):
index = x_test.index
x_test = torch.from_numpy(x_test.values).float()
x_test = x_test.to(device)
x_test = x_test.to(self.device)
self.sfm_model.eval()
with torch.no_grad():
if device != 'cpu':
if self.device != 'cpu':
preds = self.sfm_model(x_test).detach().cpu().numpy()
else:
preds = self.sfm_model(x_test).detach().numpy()