From 752f17e51e434015ec6456725a5b2b579b1d85ff Mon Sep 17 00:00:00 2001 From: lwwang1995 Date: Mon, 7 Dec 2020 10:15:11 +0800 Subject: [PATCH] Fix SFM bug. --- qlib/contrib/model/pytorch_sfm_ts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qlib/contrib/model/pytorch_sfm_ts.py b/qlib/contrib/model/pytorch_sfm_ts.py index e375e1c8d..3a1167fc5 100644 --- a/qlib/contrib/model/pytorch_sfm_ts.py +++ b/qlib/contrib/model/pytorch_sfm_ts.py @@ -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,