Commit Graph

2 Commits

Author SHA1 Message Date
qihuanye
995cd8cfec 优化 jepa.py 中通用 rollout 热路径:批量预编码动
作、移除循环内
  torch.cat,并为 history_size==1 与环形缓冲区更新
  添加更轻量实现; 收益不大
2026-04-09 11:57:09 +00:00
qihuanye
9e2407cdc4 Wrap eval inference in torch.inference_mode 2026-04-09 11:11:07 +00:00