Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "44b17d2bfa0073e012679152421c0b69671d380e"
Commit 24e41ffe authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 339598839
parent dadcf465
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
official/vision/beta/configs/semantic_segmentation.py
View file @ 24e41ffe
......@@ -39,6 +39,7 @@ class DataConfig(cfg.DataConfig):
groundtruth_padded_size: List[int] = dataclasses.field(default_factory=list)
aug_scale_min: float = 1.0
aug_scale_max: float = 1.0
drop_remainder: bool = True
@dataclasses.dataclass
......@@ -138,7 +139,8 @@ def seg_deeplabv3_pascal() -> cfg.ExperimentConfig:
is_training=False,
global_batch_size=eval_batch_size,
resize_eval_groundtruth=False,
groundtruth_padded_size=[512, 512]),
groundtruth_padded_size=[512, 512],
drop_remainder=False),
# resnet50
init_checkpoint='gs://cloud-tpu-checkpoints/vision-2.0/deeplab/deeplab_resnet50_imagenet/ckpt-62400',
init_checkpoint_modules='backbone'),
......
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