How to save part of model parameters?

I’m using vgg16 backbone on my mode and I want to save just this fine-tuned vgg16 parameters.
how can i do that?

Hi, you can do smth like{
        "epoch": epoch + 1,
        "model_state_dict": model.state_dict(),
        "optimizer": optimizer.state_dict(),
        "loss": loss
    }, model_path)

but specify only parameters you want to save

1 Like

how can i specify parameters that i want?

Just pass to save the parameters you want. E.g.

"my_param": parameters,