kozistr / pytorch_optimizer

optimizer & lr scheduler & loss function collections in PyTorch

Home Page:https://pytorch-optimizers.readthedocs.io/en/latest/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add WeLore

rotem154154 opened this issue · comments

Hi,

I really appreciate your work. For me, GaLore has worked the best out of all the optimizers. Could you please consider adding WeLore? It looks promising: https://arxiv.org/abs/2407.11239

Thank you!

thanks i'll check it later!

@rotem154154 sorry for the late reply. I just checked out the paper and it looks like it's not an optimizer, more like some tools for rank reduction. Also, their implementation only supports the specific parameters depending on the architecture such as LLaMa2 7/13B.

So, I guess it'd be good to wait until they support the model-agnostic parameters and make it easier to use.

I'll reopen this issue when the time comes.