How can I copy params to the classification model from discriminator of DCGAN

nick_debu · November 17, 2017, 2:12am

I wan to copy the discriminator model of DCGAN to Classification model.

smth · November 17, 2017, 5:16pm

get the state_dict with model.state_dict(). It is a dictionary of parameters with names.

Now filter the weights that you want to copy from the dictionary and you can call model.load_state_dict(my_modified_dict) on your newly constructed classification model.