mxnet arg_params, aux_params

最新推荐文章于 2022-09-18 09:16:15 发布

AI算法网奇

最新推荐文章于 2022-09-18 09:16:15 发布

阅读量2.1k

点赞数 1

分类专栏： mxnet

本文链接：https://blog.csdn.net/jacke121/article/details/115761105

版权

mxnet 专栏收录该内容

58 篇文章 3 订阅

订阅专栏

本文讨论了如何在_mxnet_中正确初始化参数和调用bind方法。重点在于理解arg_params和aux_params的空值问题，何时在load_pretrain和bind阶段设置参数，以及如何通过load_checkpoint加载预训练模型并确保参数在后续操作中可用。

摘要由CSDN通过智能技术生成

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
    #step2
    mod_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=True
    mod_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