How to train only middle component?

[crcrpar (Masaki Kozuki)]

Hi,

I think there are a few related threads

• Update only sub-elements of weights - autograd - PyTorch Forums
• Passing a subset of the parameters to an optimizer equivalent to setting requires_grad of subset only to True? - autograd - PyTorch Forums