11 lines
225 B
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}
|