Commit Graph

5 Commits

Author SHA1 Message Date
qihuanye
25e4ddb628 继续做了通用性能优化,重点从 jepa.py 热路径转到实际的 stable_worldmodel
solver/policy 边界:去掉 CEM 每轮 cpu().tolist() 和结果过早回 CPU,把
  plan/warm-start 保持在 GPU,只在 env.step 前最后一步转成 numpy,同时补
  了输入张量的 contiguous 处理;
2026-04-09 12:33:50 +00:00
qihuanye
20ffb3492b Disable Gym passive checker by default in stable_worldmodel env creation 2026-04-09 11:11:07 +00:00
qihuanye
0f85e39690 Reduce evaluation overhead with parallel video saving 2026-04-08 13:56:57 +00:00
qihuanye
85795bd91d Vectorize image preprocessing in stable_worldmodel policy 2026-04-08 13:48:19 +00:00
qihuanye
12ba4f4352 Optimize CEM input transfers before sample expansion 2026-04-08 13:01:24 +00:00