From 8101904f9efbfd30d05dfb3bc5863be444f8f588 Mon Sep 17 00:00:00 2001 From: Maple-YZ Date: Tue, 14 Apr 2026 06:16:43 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E9=BB=98=E8=AE=A4=E8=BF=90?= =?UTF-8?q?=E8=A1=8C=E6=A8=A1=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- training/registry.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/training/registry.py b/training/registry.py index 94669e8..2531c1d 100644 --- a/training/registry.py +++ b/training/registry.py @@ -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"]