2feab9beb8c59f05ca2e1938f484b6bce638af47,models/experimental/mask_rcnn/mask_rcnn_model.py,,_model_fn,#Any#Any#Any#Any#Any#Any#,39

Before Change

  host_call = None
  if mode == tf.estimator.ModeKeys.TRAIN:
    optimizer = tf.train.MomentumOptimizer(
        learning_rate, momentum=params["momentum"])
    optimizer = tf.contrib.tpu.CrossShardOptimizer(optimizer)

    // Batch norm requires update_ops to be added as a train_op dependency.

After Change

  host_call = None
  if mode == tf.estimator.ModeKeys.TRAIN:
    optimizer = create_optimizer(learning_rate, params)
    optimizer = tf.contrib.tpu.CrossShardOptimizer(optimizer)

    if not params["resnet_checkpoint"]:
      scaffold_fn = None

Instances