diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-07 10:41:13 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-07 10:46:59 -0700 |
| commit | 14a7b8aee0ddec4e3431f7161e9b976e7157aef7 (patch) | |
| tree | fba3119efac7d523926d131e7bcc85b7811ef434 /tensorflow/contrib/layers | |
| parent | ff3de98d531c23a574a3531d13c47ec3c27543e1 (diff) | |
Set depth_multiplier default value, so that one can use separable_conv2d and conv2d interchangeably (conv2d does not have depth_multiplier arg).
For example:
if use_separable:
conv_op = separable_conv2d
else:
conv_op = conv2d
net = conv_op(....)
Most common usage of separable conv use depth_multiplier=1, e.g. in MobileNet and Xception.
PiperOrigin-RevId: 207741004
Diffstat (limited to 'tensorflow/contrib/layers')
| -rw-r--r-- | tensorflow/contrib/layers/python/layers/layers.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/layers/python/layers/layers.py b/tensorflow/contrib/layers/python/layers/layers.py index fa334070ad..6250f88529 100644 --- a/tensorflow/contrib/layers/python/layers/layers.py +++ b/tensorflow/contrib/layers/python/layers/layers.py @@ -2660,7 +2660,7 @@ def separable_convolution2d( inputs, num_outputs, kernel_size, - depth_multiplier, + depth_multiplier=1, stride=1, padding='SAME', data_format=DATA_FORMAT_NHWC, |