Files
lewm/config/eval/solver/cem.yaml

11 lines
225 B
YAML

_target_: stable_worldmodel.solver.CEMSolver
model: ???
batch_size: 16
# Original defaults: num_samples=300, n_steps=30, topk=30, batch_size=8.
num_samples: 300
var_scale: 1.0
n_steps: 30
topk: 8
device: "cuda"
seed: ${seed}