28 lines
890 B
Python
28 lines
890 B
Python
|
import os.path
|
||
|
|
|
||
|
|
from scenarionet import SCENARIONET_DATASET_PATH
|
||
|
|
from scenarionet_training.scripts.train_pg import config
|
||
|
|
from scenarionet_training.train_utils.utils import eval_ckpt
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
# Merge all evaluate script
|
||
|
|
# 10/15/20/26/30/31/32
|
||
|
|
ckpt_path = "C:\\Users\\x1\\Desktop\\checkpoint_330\\checkpoint-330"
|
||
|
|
scenario_data_path = os.path.join(SCENARIONET_DATASET_PATH, "pg_2000")
|
||
|
|
num_scenarios = 2000
|
||
|
|
start_scenario_index = 0
|
||
|
|
horizon = 600
|
||
|
|
render = False
|
||
|
|
explore = True # PPO is a stochastic policy, turning off exploration can reduce jitter but may harm performance
|
||
|
|
log_interval = 2
|
||
|
|
|
||
|
|
eval_ckpt(config,
|
||
|
|
ckpt_path,
|
||
|
|
scenario_data_path,
|
||
|
|
num_scenarios,
|
||
|
|
start_scenario_index,
|
||
|
|
horizon,
|
||
|
|
render,
|
||
|
|
explore,
|
||
|
|
log_interval)
|