Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 454684592
parent 969a3f34
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/tasks/maskrcnn.py
View file @ 99838089
...@@ -122,7 +122,8 @@ class MaskRCNNTask(base_task.Task): ...@@ -122,7 +122,8 @@ class MaskRCNNTask(base_task.Task):
self, self,
params: exp_cfg.DataConfig, params: exp_cfg.DataConfig,
input_context: Optional[tf.distribute.InputContext] = None, input_context: Optional[tf.distribute.InputContext] = None,
dataset_fn: Optional[dataset_fn_lib.PossibleDatasetType] = None): dataset_fn: Optional[dataset_fn_lib.PossibleDatasetType] = None
) -> tf.data.Dataset:
"""Build input dataset.""" """Build input dataset."""
decoder_cfg = params.decoder.get() decoder_cfg = params.decoder.get()
if params.decoder.type == 'simple_decoder': if params.decoder.type == 'simple_decoder':
......
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