liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 16:26:55 +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
committed by you-n-g
parent da1f4db968
commit e71e2f941c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -352,7 +352,8 @@ 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()),
} }
torch.save(best_params, self.logdir + "/model.bin") if self.logdir is not None:
torch.save(best_params, self.logdir + "/model.bin")
else: else:
stop_rounds += 1 stop_rounds += 1
if stop_rounds >= self.early_stop: if stop_rounds >= self.early_stop: