1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-12 23:36:54 +08:00

Add random seed.

This commit is contained in:
lwwang1995
2020-11-30 20:51:19 +08:00
committed by you-n-g
parent d6ff62b48c
commit e0dad0e85a
5 changed files with 19 additions and 0 deletions

View File

@@ -282,6 +282,9 @@ class SFM(Model):
)
)
np.random.seed(self.seed)
torch.manual_seed(self.seed)
self.sfm_model = SFM_Model(
d_feat=self.d_feat,
output_dim=self.output_dim,