Pytorch 提取权重等参数 写入Excel

Pytorch 提取权重等参数 写入Excel表

标签 ： Pytorch

Topic : 网络参数导出

时间 ：2022 / 5 / 27

---

写在最前：

​ 最近有在做量化相关的东西，不确定是不是我这边没设置好怎么，量化后只给出了相应层的s、z值，这里就需要将网络中的参数导出进行再处理。

• 网络保存格式 ： pth
• 参数保存格式 ： 字典 dict

问题：

​ 直接打印某层参数或转换为字符串写入txt中 ：当参数多时，显示不完全，无法得到层内所有参数。

• 整体思路 ： 将字典中tensor类型变量转换成numpy array形式,然后存储到表格中。

需要用到的Python库：

• pytorch
• pandas
• numpy
• openpyxl

源代码参考自 ： pytorch 提取权重_获取Pytorch中间某一层权重或者特征的例子_闲吟客的博客-CSDN博客

## 导入必要的库
import torch
import pandas as pd
import numpy as np

## 参数存入parm字典中，同时tensor类型变量转换成numpy array