Hi,
I’m currently loading the pretrained parameters from the VGG-16 network into the first stage of my unet implementation. I’m downloading the parameters with the function model.zoo.load_url() and then loading the parameters into the model using the function load_state_dict(). However, even though all the downloaded parameters have requried_true = False, this is not updated in the model (they all have required_grad = True). Is there a way to get round this or do I have to manually set each param to false after the loading the dictionary.
Thank you for any help in advance