Skip to content

GitLab

Commit e0c3b06e authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 389672355
parent 4d2ac2df
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/vision/beta/modeling/decoders/aspp.py
View file @ e0c3b06e
...@@ -173,6 +173,7 @@ def build_aspp_decoder( ...@@ -173,6 +173,7 @@ def build_aspp_decoder(
level=decoder_cfg.level, level=decoder_cfg.level,
dilation_rates=decoder_cfg.dilation_rates, dilation_rates=decoder_cfg.dilation_rates,
num_filters=decoder_cfg.num_filters, num_filters=decoder_cfg.num_filters,
use_depthwise_convolution=decoder_cfg.use_depthwise_convolution,
pool_kernel_size=decoder_cfg.pool_kernel_size, pool_kernel_size=decoder_cfg.pool_kernel_size,
dropout_rate=decoder_cfg.dropout_rate, dropout_rate=decoder_cfg.dropout_rate,
use_sync_bn=norm_activation_config.use_sync_bn, use_sync_bn=norm_activation_config.use_sync_bn,
......
official/vision/beta/modeling/factory.py
View file @ e0c3b06e
...@@ -333,6 +333,7 @@ def build_segmentation_model( ...@@ -333,6 +333,7 @@ def build_segmentation_model(
num_convs=head_config.num_convs, num_convs=head_config.num_convs,
prediction_kernel_size=head_config.prediction_kernel_size, prediction_kernel_size=head_config.prediction_kernel_size,
num_filters=head_config.num_filters, num_filters=head_config.num_filters,
use_depthwise_convolution=head_config.use_depthwise_convolution,
upsample_factor=head_config.upsample_factor, upsample_factor=head_config.upsample_factor,
feature_fusion=head_config.feature_fusion, feature_fusion=head_config.feature_fusion,
low_level=head_config.low_level, low_level=head_config.low_level,
......
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