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

Fixed issue #943 about TCTS init_fore_model (#1047)

Browse Source 
p.init_fore_model = False -> p.requires_grad = False
This commit is contained in:
Tuozhen Liu
2022-04-14 11:23:08 +08:00
committed by GitHub
parent 87926513cb
commit 30b531086c
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_tcts.py
View File

@@ -145,7 +145,7 @@ class TCTS(Model):
init_fore_model = copy.deepcopy(self.fore_model) init_fore_model = copy.deepcopy(self.fore_model)
for p in init_fore_model.parameters(): for p in init_fore_model.parameters():
p.init_fore_model = False p.requires_grad = False
self.fore_model.train() self.fore_model.train()
self.weight_model.train() self.weight_model.train()