Breaking News: Grepper is joining You.com. Read the official announcement!

state_dict()

coder answered on April 4, 2022 Popularity 8/10 Helpfulness 5/10

answer state_dict()

state_dict()

Comment

In order to access a model's parameters in pytorch:
state_dict and  parameters()
The parameters() only gives the module parameters i.e. weights and biases.
state_dict returns a dictionary containing a whole state of the module.
model.state_dict().keys()

Popularity 8/10 Helpfulness 5/10 Language python

Source: stackoverflow.com

Tags: python

Link to this answer
Share Copy Link