Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

state_dict()

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()
Source by stackoverflow.com #
 
PREVIOUS NEXT
Tagged:
ADD COMMENT
Topic
Name
8+1 =