qihuanye 995cd8cfec 优化 jepa.py 中通用 rollout 热路径：批量预编码动 ...

作、移除循环内
  torch.cat，并为 history_size==1 与环形缓冲区更新
  添加更轻量实现； 收益不大

2026-04-09 11:57:09 +00:00

42 KiB

Raw Blame History

View Raw