胶子序列模型:参数返回空集合

2024-06-07 18:37:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用下面的代码创建一个连续胶子模型。由于某些原因,属性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_ (

)

Tags: 代码模型buildaddnet属性原因nn
1条回答
网友
1楼 · 发布于 2024-06-07 18:37:36

使用Sequential.collect_params(),它不仅收集这个Block参数,而且还收集所有子项(例如层)

相关问题 更多 >

    热门问题