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

improve pretrain

Browse Source
This commit is contained in:
Dong Zhou
2021-07-28 10:07:35 +08:00
committed by you-n-g
parent 5b7b48e376
commit a7c41b6969
4 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_tra.py
View File

@@ -392,7 +392,7 @@ class TRAModel(Model):
self.logger.info("reset TRA")
self.tra.reset_parameters() # reset both router and predictors
self.optimizer = optim.Adam(self.tra.parameters(), lr=self.lr) # optimize TRA only
self.optimizer = optim.Adam(list(self.model.parameters()) + list(self.tra.parameters()), lr=self.lr)
self.logger.info("training...")
best_score, _ = self._fit(train_set, valid_set, test_set, evals_result, start_epoch=epoch, is_pretrain=False)