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

8
scenarionet_training/scripts/local_test.py
View File

@@ -1,7 +1,7 @@
import os.path
from metadrive.envs.scenario_env import ScenarioEnv
from metadrive.envs.gym_wrapper import GymEnvWrapper
from metadrive.envs.gym_wrapper import createGymWrapper
from scenarionet import SCENARIONET_REPO_PATH, SCENARIONET_DATASET_PATH
from scenarionet_training.train_utils.multi_worker_PPO import MultiWorkerPPO
@@ -14,8 +14,8 @@ if __name__ == '__main__':
stop = int(100_000_000)
config = dict(
env=GymEnvWrapper,
env_config=dict(inner_class=ScenarioEnv, inner_config=dict(
env=createGymWrapper(ScenarioEnv),
env_config=dict(
# scenario
start_scenario_index=0,
num_scenarios=32,
@@ -34,7 +34,7 @@ if __name__ == '__main__':
# training
horizon=None,
)),
),
# # ===== Evaluation =====
evaluation_interval=2,