Skip to content

GitLab

Commit f123edc0 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 366391425
parent bab70e6b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/modeling/backbones/resnet_deeplab.py
View file @ f123edc0
...@@ -77,8 +77,8 @@ class DilatedResNet(tf.keras.Model): ...@@ -77,8 +77,8 @@ class DilatedResNet(tf.keras.Model):
output_stride: An `int` of output stride, ratio of input to output output_stride: An `int` of output stride, ratio of input to output
resolution. resolution.
input_specs: A `tf.keras.layers.InputSpec` of the input tensor. input_specs: A `tf.keras.layers.InputSpec` of the input tensor.
stem_type: A `str` of stem type. Can be `standard` or `deeplab`. `deeplab` stem_type: A `str` of stem type. Can be `v0` or `v1`. `v1` replaces 7x7
replaces 7x7 conv by 3 3x3 convs. conv by 3 3x3 convs.
se_ratio: A `float` or None. Ratio of the Squeeze-and-Excitation layer. se_ratio: A `float` or None. Ratio of the Squeeze-and-Excitation layer.
init_stochastic_depth_rate: A `float` of initial stochastic depth rate. init_stochastic_depth_rate: A `float` of initial stochastic depth rate.
multigrid: A tuple of the same length as the number of blocks in the last multigrid: A tuple of the same length as the number of blocks in the last
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment