MapleL
/
ctm-dqn
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修改默认运行模型

This commit is contained in:
Zihan Ye 2026-04-14 06:16:43 +08:00
parent e3e4cff22d
commit 8101904f9e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
training/registry.py
View File

@ -16,8 +16,8 @@ from training.train_tcamappo import train_sumo_tcamappo
from training.train_td3 import train_sumo_td3
DEFAULT_MODELS: List[str] = ["ppo"]
# DEFAULT_MODELS: List[str] = ["ppo", "gpro", "mappo", "tcamappo", "dcmappo", "dqn", "madqn", "ddqn", "ddpg", "sac", "td3"]
# DEFAULT_MODELS: List[str] = ["ppo"]
DEFAULT_MODELS: List[str] = ["ppo", "gpro", "mappo", "tcamappo", "dcmappo", "dqn", "madqn", "ddqn", "ddpg", "sac", "td3"]
ALL_MODELS: List[str] = ["ppo", "gpro", "appo", "mappo", "tcamappo", "dcmappo", "dqn", "madqn", "ddqn", "ddpg", "sac", "td3", "sctd3"]