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