How can I get the weights of a PyTorch model?

Question

I've just built a PyTorch model and want to get the weights of each layer.

What is the best way to do it?

Sunday, December 27, 2020 4:50 PM

2 votesJOJack Oliver297.00

1 Answers

Couldn't find what you were looking for?and we will find an expert to answer.

How helpful was this page?

John Wilson · Accepted Answer

It's easy to do if you use PyTorch.

for name, param in model.named_parameters():
    # layer name, layer weights        
    params.append(param)