fix: exp14 finetune eval uses make_env_base (runtime throttle floor) instead of removed make_env
This commit is contained in:
parent
6c5623e881
commit
f3c89116ee
|
|
@ -127,7 +127,8 @@ def make_env_base(base_throttle=0.2, throttle_floor=None):
|
||||||
from stable_baselines3.common.utils import set_random_seed
|
from stable_baselines3.common.utils import set_random_seed
|
||||||
|
|
||||||
def evaluate_model(model, throttle_min, sets=EVAL_EPISODES):
|
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 = []
|
results = []
|
||||||
for s in range(sets):
|
for s in range(sets):
|
||||||
obs = env.reset()
|
obs = env.reset()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue