pytorch查看网络权重

model.state_dict()返回一个字典包含权重

 

model.state_dict().keys()

返回  odict_keys(['fc1.weight', 'fc1.bias', 'fc2.weight', 'fc2.bias', 'fc3.weight', 'fc3.bias'])

 

model.fc1.weight 返回第一层的权重