完善项目目录结构

2026-02-03 16:24:15 +08:00
parent 21c046aef0
commit 03dee0205a
27 changed files with 558 additions and 568 deletions
--- a/train_magail.py
+++ b/train_magail.py
@@ -9,7 +9,7 @@ import argparse
 import signal
 import sys
 from torch.utils.data import DataLoader
-from dataset.magail_dataset import MAGAILExpertDataset
+from dataset.loader import MAGAILExpertDataset
 from Env.bc_env import BCScenarioEnv

 # --- Networks ---