huangfu/scenarionet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rebuttal (#15)

Browse Source 
* pg+nuplan train

* Need map

* use gym wrapper

* use createGymWrapper

* doc

* use all scenarios!

* update 80000 scenario

* train script
This commit is contained in:
Quanyi Li
2023-08-08 17:33:02 +01:00
committed by GitHub
parent e9c1419a91
commit 64bf9811b9
12 changed files with 274 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scenarionet_training/train_utils/utils.py
View File

@@ -9,7 +9,7 @@ import numpy as np
import tqdm
from metadrive.constants import TerminationState
from metadrive.envs.scenario_env import ScenarioEnv
from metadrive.envs.gym_wrapper import GymEnvWrapper
from metadrive.envs.gym_wrapper import createGymWrapper
from ray import tune
from ray.tune import CLIReporter
@@ -292,7 +292,7 @@ def eval_ckpt(config,
episodes_to_evaluate_curriculum=num_scenarios,
data_directory=scenario_data_path,
use_render=render))
env = GymEnvWrapper(dict(inner_class=ScenarioEnv, inner_config=env_config))
env = createGymWrapper(ScenarioEnv)(env_config)
super_data = defaultdict(list)
EPISODE_NUM = env.config["num_scenarios"]