11 lines
223 B
YAML
11 lines
223 B
YAML
_target_: stable_worldmodel.solver.CEMSolver
|
|
model: ???
|
|
batch_size: 8
|
|
# Original defaults: num_samples=300, n_steps=30, topk=30, batch_size=8.
|
|
num_samples: 64
|
|
var_scale: 1.0
|
|
n_steps: 10
|
|
topk: 8
|
|
device: "cuda"
|
|
seed: ${seed}
|