赞
踩
标签 : Pytorch
Topic : 网络参数导出
时间 :2022 / 5 / 27
写在最前:
最近有在做量化相关的东西,不确定是不是我这边没设置好怎么,量化后只给出了相应层的s、z值,这里就需要将网络中的参数导出进行再处理。
问题:
直接打印某层参数或转换为字符串写入txt中 :当参数多时,显示不完全,无法得到层内所有参数。
需要用到的Python库:
源代码参考自 : pytorch 提取权重_获取Pytorch中间某一层权重或者特征的例子_闲吟客的博客-CSDN博客
## 导入必要的库
import torch
import pandas as pd
import numpy as np
## 参数存入parm字典中,同时tensor类型变量转换成numpy array
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。