huggingface / alignment-handbook

Robust recipes to align language models with human and AI preferences

Home Page:https://huggingface.co/HuggingFaceH4

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is there a way to freeze some layers of a model ?

shamanez opened this issue · comments

Can we follow the normal way of:

for param in model.base_model.parameters():
    param.requires_grad = False