Question

我正在使用下面的代码创建顺序胶粘剂模型。由于某种原因，属性params返回一个空集合。

def build_net():
    net = gluon.nn.Sequential()
    with net.name_scope():
        net.add(gluon.nn.Dense(32, activation='relu'))
        net.add(gluon.nn.Dense(32, activation='relu'))
        net.add(gluon.nn.Dense(1))

    net.collect_params().initialize(mx.init.Normal(sigma=.1))
    return net

net_1 = build_net() 
print(net_1.params)

输出：

sequential0_ (

)

Answer 1

使用z，它不仅收集此Sequential.collect_params()参数，而且还收集所有子项（例如Layers）的数据。

Answer 2

使用collect_params()返回一个包含该块及其所有子参数的ParameterDict

def build_net():
    net = gluon.nn.Sequential()
    with net.name_scope():
        net.add(gluon.nn.Dense(32, activation='relu'))
        net.add(gluon.nn.Dense(32, activation='relu'))
        net.add(gluon.nn.Dense(1))

    net.initialize(mx.init.Normal(sigma=.1))
    return net

net_1 = build_net() 
print(net_1.collect_params())

gluon顺序模型：参数返回空集合

2 个答案: