gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-02 18:07:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support run one step for debugging (#99)

Browse Source
This commit is contained in:
you-n-g
2024-07-23 19:00:59 +08:00
committed by GitHub
parent 44169507b2
commit 40d108687c
2 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/app/qlib_rd_loop/model_w_sc.py
+3 -3
View File
@@ -67,20 +67,20 @@ class ModelLoop(LoopBase, metaclass=LoopMeta):
self.trace.hist.append((prev_out["propose"],prev_out["running"] , feedback))
def main(path=None):
def main(path=None, step_n=None):
"""
You can continue running session by
.. code-block:: python
dotenv run -- python rdagent/app/qlib_rd_loop/model_w_sc.py $LOG_PATH/__session__/1/0_propose
dotenv run -- python rdagent/app/qlib_rd_loop/model_w_sc.py $LOG_PATH/__session__/1/0_propose --step_n 1 # `step_n` is a optional paramter
"""
if path is None:
model_loop = ModelLoop()
else:
model_loop = ModelLoop.load(path)
model_loop.run()
model_loop.run(step_n=step_n)
if __name__ == "__main__":