Skip to content

GitLab

Commit 0f091a60 authored by Rebecca Chen's avatar Rebecca Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 458331830
parent b6780b59
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/projects/volumetric_models/modeling/decoders/unet_3d_decoder.py
View file @ 0f091a60
...@@ -95,7 +95,7 @@ class UNet3DDecoder(tf.keras.Model): ...@@ -95,7 +95,7 @@ class UNet3DDecoder(tf.keras.Model):
channel_dim = 1 channel_dim = 1
# Build 3D UNet. # Build 3D UNet.
inputs = self._build_input_pyramid(input_specs, model_id) inputs = self._build_input_pyramid(input_specs, model_id) # pytype: disable=wrong-arg-types # dynamic-method-lookup
# Add levels with up-convolution or up-sampling. # Add levels with up-convolution or up-sampling.
x = inputs[str(model_id)] x = inputs[str(model_id)]
......
official/vision/modeling/maskrcnn_model.py
View file @ 0f091a60
...@@ -158,7 +158,7 @@ class MaskRCNNModel(tf.keras.Model): ...@@ -158,7 +158,7 @@ class MaskRCNNModel(tf.keras.Model):
matched_gt_classes=intermediate_outputs['matched_gt_classes'], matched_gt_classes=intermediate_outputs['matched_gt_classes'],
gt_masks=gt_masks, gt_masks=gt_masks,
training=training) training=training)
model_outputs.update(model_mask_outputs) model_outputs.update(model_mask_outputs) # pytype: disable=attribute-error # dynamic-method-lookup
return model_outputs return model_outputs
def _get_backbone_and_decoder_features(self, images): def _get_backbone_and_decoder_features(self, images):
......
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