liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-18 01:44:34 +08:00
Code Activity

fix tra when logdir is None

Browse Source
This commit is contained in:
Dong Zhou
2021-08-02 03:57:14 -07:00
parent da1f4db968
commit 0483406c12
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -352,6 +352,7 @@ class TRAModel(Model):
"model": copy.deepcopy(self.model.state_dict()), "model": copy.deepcopy(self.model.state_dict()),
"tra": copy.deepcopy(self.tra.state_dict()), "tra": copy.deepcopy(self.tra.state_dict()),
} }
if self.logdir is not None:
torch.save(best_params, self.logdir + "/model.bin") torch.save(best_params, self.logdir + "/model.bin")
else: else:
stop_rounds += 1 stop_rounds += 1