我正在使用下面的代码创建一个连续胶子模型。由于某些原因,属性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_ (
)
使用
Sequential.collect_params()
,它不仅收集这个Block
参数,而且还收集所有子项(例如层)相关问题 更多 >
编程相关推荐