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

Fix SFM bug.

Browse Source
This commit is contained in:
lwwang1995
2020-12-07 10:15:11 +08:00
committed by you-n-g
parent 8d42092a7e
commit 752f17e51e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -122,7 +122,7 @@ class SFM(Model):
np.random.seed(self.seed) np.random.seed(self.seed)
torch.manual_seed(self.seed) torch.manual_seed(self.seed)
self.SFM_model = SFMModel( self.SFM_model = SFM_Model(
d_feat=self.d_feat, d_feat=self.d_feat,
hidden_size=self.hidden_size, hidden_size=self.hidden_size,
num_layers=self.num_layers, num_layers=self.num_layers,