Training with frozen backbone (head only)

[Chrisguagliano]

Hi,

Perhaps I've missed it or unsure of how to implement it, but is there a built in option to freeze the backbone and train only the heads (classification and regression networks)?

Thank you!
E

[rwightman]

@esquirel123 no, currently no builtin interface for that (on the list of possible TODOs). You'll have to manually iterate over the modules and set requires_grad=False, set BN mode, etc.