Commit Graph

3 Commits

Author SHA1 Message Date
qihuanye
28f2fba0e8 加入一个提前停止的机制 还有减少环境步中间步骤传递至cpu 2026-05-18 00:48:59 +08:00
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
85795bd91d Vectorize image preprocessing in stable_worldmodel policy 2026-04-08 13:48:19 +00:00