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 and update run_all_model

Browse Source
This commit is contained in:
Jactus
2020-12-01 11:54:26 +08:00
parent bb5f3cb33d
commit 77884db3a5
12 changed files with 42 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -217,7 +217,7 @@ class SFM(Model):
loss="mse",
optimizer="gd",
GPU="0",
seed=0,
seed=None,
**kwargs
):
# Set logger.
@@ -282,8 +282,9 @@ class SFM(Model):
)
)
np.random.seed(self.seed)
torch.manual_seed(self.seed)
if self.seed is not None:
np.random.seed(self.seed)
torch.manual_seed(self.seed)
self.sfm_model = SFM_Model(
d_feat=self.d_feat,