资讯详情

从小白入手,从已经训练好的模型中取出weight权重参数绘制柱状图

从模型pt取出文件中的权重参数,绘制柱状图

  • 一、详细步骤
      • 第一步:读取pt变量中的文件content中,并查看content字典中键有以下两种方。
        • 1.1 加载pt查看文件后content中的key
        • 1.2 pt文件保存txt文件:成功!
        • 1.3 过滤content中,取出model中含有weight关键字的key
        • 获取model中某一key的shape
    • 1.3 获取content中的(model)key对应的value
    • 1.4 取出model中卷积层值
  • 二、从模型pt取出文件中的权重参数,存储数据excel表格中-完整代码附上
      • 结果图-成功!

标签: 温度传感器wkl

锐单商城拥有海量元器件数据手册IC替代型号,打造 电子元器件IC百科大全!

锐单商城 - 一站式电子元器件采购平台