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:
parent
50d816cbc5
commit
91c9366589
@ -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:
|
||||||
|
Loading…
Reference in New Issue
Block a user