3a4ac1eee9039eee0f0a3aad1e55c1394446b7ba,official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py,,run,#Any#,99

Before Change

  per_epoch_steps, train_epochs, eval_steps = get_num_train_iterations(
      flags_obj)
  steps_per_loop = min(flags_obj.steps_per_loop, per_epoch_steps)

  logging.info(
      "Training %d epochs, each epoch has %d steps, "
      "total steps: %d; Eval %d steps", train_epochs, per_epoch_steps,

After Change

      steps_per_loop=steps_per_loop,
      checkpoint_manager=checkpoint_manager,
      summary_interval=summary_interval,
      eval_summary_dir=os.path.join(flags_obj.model_dir, "eval"))

  time_callback.on_train_begin()
  if not flags_obj.skip_eval:

Instances