liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-06 05:51:17 +08:00
Code Activity

fix pkl file not loading in StaticDataLoader (#1896)

Browse Source 
* fix pkl file not loading in StaticDataLoader

* resolve hard code

* resolve hard code
This commit is contained in:
Linlang
2025-03-18 16:05:24 +08:00
committed by GitHub
parent 82f1ef2def
commit 4d621bff99
9 changed files with 20 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/rl/test_trainer.py
View File

@@ -194,7 +194,7 @@ def test_trainer_checkpoint():
assert (output_dir / "002.pth").exists()
assert os.readlink(output_dir / "latest.pth") == str(output_dir / "002.pth")
trainer.load_state_dict(torch.load(output_dir / "001.pth"))
trainer.load_state_dict(torch.load(output_dir / "001.pth", weights_only=False))
assert trainer.current_iter == 1
assert trainer.current_episode == 100