def dense_layer(input, n_units, name, network_weights, nonlinearity=None, bn=False):
layer = DenseLayer(input, num_units=n_units, nonlinearity=nonlinearity, name=name,
W=get_W(network_weights, name), b=get_b(network_weights, name))
if bn:
layer = batch_norm(layer)
return layer
评论列表
文章目录