CarperAI/trlx

Collapse reference+learner hydra heads when using LoRa

Open

#320 创建于 2023年2月21日

在 GitHub 查看
 (6 评论) (2 反应) (0 负责人)Python (450 fork)batch import
contributions welcomefeature requestgood first issue

仓库指标

Star
 (4,184 star)
PR 合并指标
 (30 天内没有已合并 PR)

描述

🚀 The feature, motivation, and pitch

With additive (delta-style) parameter-efficient tuning methods such as LoRa, we should be able to make a slightly more mem-efficient hydra architecture by using a single block that does ~frozen_head + tunable_weights for the learner/policy head's fwd-pass and simply frozen_head for the reference, instead of maintaining 2x heads.

CC @LouisCastricato and @cat-state for pointing this out

Alternatives

No response

Additional context

No response

贡献者指南