Including batch_norm as the normalizer function by default, as mentioned in function description (#9652)

* fixed separable_conv2d description error and included batch_norm by default as mentioned in description

* Update layers.py

* Update layers.py
This commit is contained in:
Kwotsin 2017-05-07 01:50:35 +08:00 committed by drpngx
parent 50d816cbc5
commit 91c9366589

View File

@ -1087,7 +1087,7 @@ def convolution2d_transpose(
"""Adds a convolution2d_transpose with an optional batch normalization layer. """Adds a convolution2d_transpose with an optional batch normalization layer.
The function creates a variable called `weights`, representing the The function creates a variable called `weights`, representing the
kernel, that is convolved with the input. If `batch_norm_params` is `None`, a kernel, that is convolved with the input. If `normalizer_fn` is `None`, a
second variable called 'biases' is added to the result of the operation. second variable called 'biases' is added to the result of the operation.
Args: Args:
@ -1847,9 +1847,9 @@ def separable_convolution2d(
This op first performs a depthwise convolution that acts separately on This op first performs a depthwise convolution that acts separately on
channels, creating a variable called `depthwise_weights`. If `num_outputs` channels, creating a variable called `depthwise_weights`. If `num_outputs`
is not None, it adds a pointwise convolution that mixes channels, creating a is not None, it adds a pointwise convolution that mixes channels, creating a
variable called `pointwise_weights`. Then, if `batch_norm_params` is None, variable called `pointwise_weights`. Then, if `normalizer_fn` is None,
it adds bias to the result, creating a variable called 'biases', otherwise it adds bias to the result, creating a variable called 'biases', otherwise,
it adds a batch normalization layer. It finally applies an activation function the `normalizer_fn` is applied. It finally applies an activation function
to produce the end result. to produce the end result.
Args: Args: