initial RL training tools and loop
This commit is contained in:
parent
51a6b7d2b5
commit
f018999da9
5 changed files with 1199 additions and 3 deletions
12
toolsets.py
12
toolsets.py
|
|
@ -90,6 +90,18 @@ TOOLSETS = {
|
|||
"includes": []
|
||||
},
|
||||
|
||||
"rl": {
|
||||
"description": "RL training tools for running reinforcement learning on Tinker-Atropos",
|
||||
"tools": [
|
||||
"rl_list_environments", "rl_select_environment",
|
||||
"rl_get_current_config", "rl_edit_config",
|
||||
"rl_start_training", "rl_check_status",
|
||||
"rl_stop_training", "rl_get_results",
|
||||
"rl_test_inference", "rl_list_runs"
|
||||
],
|
||||
"includes": []
|
||||
},
|
||||
|
||||
# Scenario-specific toolsets
|
||||
|
||||
"debugging": {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue