model.state_dict()返回一个字典包含权重
model.state_dict().keys()
返回 odict_keys(['fc1.weight', 'fc1.bias', 'fc2.weight', 'fc2.bias', 'fc3.weight', 'fc3.bias'])
model.fc1.weight 返回第一层的权重
posted on 2019-06-05 10:50 那抹阳光1994 阅读(7827) 评论(0) 编辑 收藏 举报
