5939963d6762d8788d4b4b8ba1a66734bf3b4219,models/official/mnasnet/mnasnet_main.py,,,#,37

Before Change

        "The model name to select models among existing MnasNet configurations."
    ))

flags.DEFINE_string(
    "mode",
    default="train_and_eval",
    help="One of {"train_and_eval", "train", "eval"}.")

flags.DEFINE_integer(
    "train_steps",
    default=437898,

After Change

        "The model name to select models among existing MnasNet configurations."
    ))

flags.DEFINE_enum("mode", "train_and_eval",
                  ["train_and_eval", "train", "eval", "export_only"],
                  "One of {"train_and_eval", "train", "eval", "export_only"}.")

flags.DEFINE_integer(
    "train_steps",
    default=437898,

Instances