def conv_layer(input, n_filters, stride, name, network_weights, nonlinearity=elu, bn=False):
layer = Conv2DLayer(input, num_filters=n_filters, filter_size=3, stride=stride, pad='same',
nonlinearity=nonlinearity, name=name, W=get_W(network_weights, name), b=get_b(network_weights, name))
if bn:
layer = batch_norm(layer)
return layer
评论列表
文章目录