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