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

修复ppo模型训练存储目录不对的问题

This commit is contained in:
Zihan Ye 2026-03-29 03:17:39 +08:00
parent 6e135e6cb8
commit 95157bc8da
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train_sumo_ppo.py
View File

@ -29,8 +29,8 @@ def train_sumo_ppo():
start_episode = 1 start_episode = 1
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
checkpoint_dir = os.path.join(train_config["checkpoint_dir"], timestamp) checkpoint_dir = os.path.join("checkpoints_sumo_ppo", timestamp)
log_dir = os.path.join(train_config["log_dir"], timestamp) log_dir = os.path.join("logs_sumo_ppo", timestamp)
os.makedirs(checkpoint_dir, exist_ok=True) os.makedirs(checkpoint_dir, exist_ok=True)
os.makedirs(log_dir, exist_ok=True) os.makedirs(log_dir, exist_ok=True)
with open(os.path.join(checkpoint_dir, "config.yaml"), "w", encoding="utf-8") as f: with open(os.path.join(checkpoint_dir, "config.yaml"), "w", encoding="utf-8") as f: