1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-03 11:00:57 +08:00

Fix SFM bug.

This commit is contained in:
lwwang1995
2020-12-07 10:15:11 +08:00
committed by you-n-g
parent 8d42092a7e
commit 752f17e51e

View File

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