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

format

Browse Source
This commit is contained in:
QuanyiLi
2023-08-08 17:51:04 +01:00
parent fac13e2e5f
commit 52ff95a078
1 changed files with 35 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scenarionet/tests/script/run_ckpt.py
View File

@@ -7,7 +7,8 @@ from scenarionet_training.scripts.train_nuplan import config
if __name__ == "__main__": if __name__ == "__main__":
initialize_ray(test_mode=False, num_gpus=1) initialize_ray(test_mode=False, num_gpus=1)
env = createGymWrapper(ScenarioEnv)({ env = createGymWrapper(ScenarioEnv)(
{
# "data_directory": AssetLoader.file_path("nuscenes", return_raw_style=False), # "data_directory": AssetLoader.file_path("nuscenes", return_raw_style=False),
"data_directory": "D:\\scenarionet_testset\\nuplan_test\\nuplan_test_w_raw", "data_directory": "D:\\scenarionet_testset\\nuplan_test\\nuplan_test_w_raw",
"use_render": True, "use_render": True,
@@ -42,7 +43,6 @@ if __name__ == "__main__":
} }
) )
# env.reset() # env.reset()
# #
# #