pytorch查看网络权重
摘要:
model.state_dict()返回一个字典包含权重 model.state_dict().keys() 返回 odict_keys(['fc1.weight', 'fc1.bias', 'fc2.weight', 'fc2.bias', 'fc3.weight', 'fc3.bias']) m 阅读全文
posted @ 2019-06-05 10:50 那抹阳光1994 阅读(7827) 评论(0) 推荐(1) 编辑