diff --git a/agent/experiments/exp14_finetune_v5.py b/agent/experiments/exp14_finetune_v5.py index 19e4ff9..d119497 100644 --- a/agent/experiments/exp14_finetune_v5.py +++ b/agent/experiments/exp14_finetune_v5.py @@ -127,7 +127,8 @@ def make_env_base(base_throttle=0.2, throttle_floor=None): from stable_baselines3.common.utils import set_random_seed def evaluate_model(model, throttle_min, sets=EVAL_EPISODES): - env = VecTransposeImage(DummyVecEnv([make_env(throttle_min)])) + # Keep base action space at 0.2 and optionally enforce a runtime floor. + env = VecTransposeImage(DummyVecEnv([make_env_base(0.2, throttle_floor=throttle_min)])) results = [] for s in range(sets): obs = env.reset()