4b7c183cfca336fcc96b5a3c45cb99eda260e8df,models/official/mask_rcnn/mask_rcnn_main.py,,,#,30

Before Change

    help="Number of workers when using collective all reduce strategy.")

// Cloud TPU Cluster Resolvers
flags.DEFINE_string(
    "tpu",
    default=None,
    help="The Cloud TPU to use for training. This should be either the name "
    "used when creating the Cloud TPU, or a grpc://ip.address.of.tpu:8470 "
    "url.")
flags.DEFINE_string(
    "gcp_project",
    default=None,
    help="Project name for the Cloud TPU-enabled project. If not specified, we "
    "will attempt to automatically detect the GCE project from metadata.")
flags.DEFINE_string(
    "tpu_zone",
    default=None,
    help="GCE zone where the Cloud TPU is located in. If not specified, we "
    "will attempt to automatically detect the GCE project from metadata.")

// TPUEstimator parameters
flags.DEFINE_integer(
    "num_cores", default=8, help="Number of TPU cores for training")
flags.DEFINE_multi_integer(
    "input_partition_dims", None,
    "A list that describes the partition dims for all the tensors.")
flags.DEFINE_integer("iterations_per_loop", 2500,
                     "Number of iterations per TPU training loop")
flags.DEFINE_bool(
    "transpose_input",
    default=True,
    help="Use TPU double transpose optimization")

// Model specific paramenters
flags.DEFINE_string(
    "eval_master", default="",
    help="GRPC URL of the eval master. Set to an appropiate value when running "
    "on CPU/GPU")
flags.DEFINE_bool("use_tpu", True, "Use TPUs rather than CPUs")
flags.DEFINE_string("model_dir", None, "Location of model_dir")
flags.DEFINE_string(
    "config", "",
    "A comma-separated k=v pairs, or a YAML config file that specifies the "
    "parameters to build, train and eval the model.")

flags.DEFINE_string("mode", "train",
                    "Mode to run: train or eval (default: train)")
flags.DEFINE_bool("eval_after_training", False, "Run one eval after the "
                  "training finishes.")
flags.DEFINE_bool("use_fake_data", False, "Use fake input.")

// For Eval mode
flags.DEFINE_integer("min_eval_interval", 180,

After Change

from configs import mask_rcnn_config

common_tpu_flags.define_common_tpu_flags()
common_hparams_flags.define_common_hparams_flags()

flags.DEFINE_string(
    "distribution_strategy",
    default="tpu",

Instances