Fixed lint and flag issues

official/resnet/keras/keras_cifar_benchmark.py

@@ -52,7 +52,7 @@ class KerasCifar10BenchmarkTests(object):
   def keras_resnet56_no_dist_strat_1_gpu(self):
     """Test keras based model with Keras fit but not distribution strategies."""
     self._setup()
-    flags.FLAGS.dist_strat_off = True
+    flags.FLAGS.turn_off_distribution_strategy = True
     flags.FLAGS.num_gpus = 1
     flags.FLAGS.data_dir = DATA_DIR
     flags.FLAGS.batch_size = 128

official/resnet/keras/keras_cifar_main.py

@@ -39,7 +39,7 @@ def learning_rate_schedule(current_epoch,
                            current_batch,
                            batches_per_epoch,
                            batch_size):
-  """Handles linear scaling rule, gradual warmup, and LR decay.
+  """Handles linear scaling rule and LR decay.
 
   Scale learning rate at epoch boundaries provided in LR_SCHEDULE by the
   provided scaling factor.