b20fc24180a7591d6e1850ebb0fc2812205bdfdf,examples/in_cluster/kubeflow/tfjob/run.py,,,#,185

Before Change

  parser.add_argument(
      "--log_dir",
      type=str,
      default=os.path.join(os.getenv("TEST_TMPDIR", "/tmp"),
                           "tensorflow/logs"),
      help="Summaries log directory")
  FLAGS, unparsed = parser.parse_known_args()

After Change

        type=str,
        default="relu"
    )
    parser.add_argument(
        "--optimizer",
        type=str,
        default="adam"
    )
    parser.add_argument(
        "--learning_rate",
        type=float,
        default=0.001

Instances