加入一个提前停止的机制还有减少环境步中间步骤传递至cpu

2026-05-18 00:48:59 +08:00
parent 113e591899
commit 28f2fba0e8
5 changed files with 138 additions and 16 deletions
--- a/config/eval/solver/cem.yaml
+++ b/config/eval/solver/cem.yaml
@@ -2,9 +2,9 @@ _target_: stable_worldmodel.solver.CEMSolver
 model: ???
 batch_size: 16
 # Original defaults: num_samples=300, n_steps=30, topk=30, batch_size=8.
-num_samples: 64
+num_samples: 300
 var_scale: 1.0
-n_steps: 10
+n_steps: 30
 topk: 8
 device: "cuda"
 seed: ${seed}