文章分类 - tensorflow和pytorch代码段对比

torch和tensorflow获取权重

摘要：torch: 1. 通过model.state_dict()输出模型结构，结构中key是权重名称，value是权重的值 2. 根据权重名称获取权重： fc_weight= model.state_dict()['fc_cls.weight']# 权重名称为：fc_cls.weight tensorf 阅读全文

posted @ 2023-09-18 18:54 一点飞鸿阅读(185) 评论(0) 推荐(0)