mxnet arg_params, aux

arg_params, aux_params

构建网络后，这个参数是空的

如果设置这个参数：

mod_new.binded=True

bind()里面有判断，调用过了，就不会再执行bind()内存了，获取参数会为空

load函数：

load("pretrain",0)

这个代码也是给arg_params赋值，

什么时候初始化?，bind的时候会操作

参数在bind里面产生

sym, arg_params, aux_params = mx.model.load_checkpoint(r"E:\jinji\AidLearning-FrameWork-master\src\facencnn\models\mv2_55\model", 0)sym_new=get_symbol(512)mod_new=mx.mod.Module(symbol=sym_new,context=[mx.cpu()])#,label_names=None)# step1 这里获取参数为空# mod_new.params_initialized=True# mod_new.get_params()data_shape_w=112# mod_new.binded=True#step2mod_new.bind(for_training=False,data_shapes=[('data',(1,3,data_shape_w,data_shape_w))])#step2 此时网络会生成参数arg_params_new=dict()aux_params_new=dict()for key in arg_params.keys():arg_params_new[key]=arg_params[key]for key in aux_params.keys():aux_params_new[key]=aux_params[key]#step3 这里获取参数，就有了mod_new.params_initialized=Truemod_new.get_params()mod_new.set_params(arg_params_new,aux_params_new,allow_missing=True)mod_new.save_checkpoint("aaaa",0)
————————————————
版权声明：本文为CSDN博主「ShellCollector」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/jacke121/article/details/115767733

mxnet arg_params, aux_params相关推荐

Windows 10 Anaconda Python 3.7 安装 MXNet GPU版
MXNet CPU版本可以直接通过pip进行安装: pip install mxnet MXNet GPU版本不推荐通过pip进行安装,推荐使用conda进行安装: conda install -c ...
mxnet 查看中间层结果
import mxnet as mx from mxnet import nd from mxnet.gluon import nnmx.cpu(), mx.gpu(), mx.gpu(0) 查看mx ...
mxnet根据相似度进行人脸样本对图片清理
清理的角度: 找出 ID (gallery) 和它的 prob 最不相似的数据 (可能是标注错误,不同ID的数据错误混杂等) 找出 ID 和别的 prob 最相似的数据 (可能是标注错误,数据 ...
mxnet可视化模型中间层feature map输出
注: model输入 112x112 保存的图片可能是白色的(这个还没有修复),但是在pycharm中运行时可以通过scientific tool窗口看到 #构造辅助函数做预处理, 注意mxnet中为 ...
mxnet is not presented
加载预训练 sym, arg_params, aux_params = mx.model.load_checkpoint(r"/mxnet/new_model", 0)sym_ne ...
insightface mxnet训练旧版
11年it研发经验,从一个会计转行为算法工程师,学过C#,c++,java,android,php,go,js,python,CNN神经网络,四千多篇博文,三千多篇原创,只为与你分享,共同成长,一起进 ...
insightface mxnet训练horovod版
依赖项: horovod Horovod是Uber开源的又一个深度学习工具,它的发展吸取了Facebook "Training ImageNet In 1 Hour" 与百度 &q ...
mxnet加载保存部分参数
def get_specify_mod(model_str,ctx,data_shpae,layer_name):_vec = model_str.split(",")prefix ...
Mxnet TensorRT
Optimizing Deep Learning Computation Graphs with TensorRT CUDA9.0或9.2 Pascal或更新架构的显卡下载并安装TensorRT库 ...

mxnet arg_params, aux_params

mxnet arg_params, aux_params相关推荐

最新文章

热门文章