Pytorch lightning callback to freeze backbone

[fstroth]

Pytorch lightning callback to freeze backbone

Describe the solution you'd like Write callback the takes an int for the number of epochs to freeze the backbone for.

Pls assign to me.

[FraPochetti]

Do you mean something like this one?

[fstroth]

Yes. For a first implementation, I would like to mimic the interface of FastAI where we freeze the backbone for n epochs and then train both. Your version seems to use a gradual approach, which could be one of the next steps.