galidor/pytorchpartiallayerfreezing

A simple code snippet for freezing only part of the layer

pytorchpartiallayerfreezing's Issues

What is the license of the project?

Thanks for this project!

I'd like to rearrange it to a Python package that can be installed through pip. I can do it only if the project is open-source (e.g., MIT license)

train model with partial frozen filters on cuda

Hi,

I am trying to freeze some filters of the conv2d layer and train the network on cuda. However, this gives error in the backward pass. Could you please help me with that ?
The error is :
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

Thanks!