9f3cd533db7695b58f95b369c16bb1b3009e4714,examples/mnist/keras/mnist_spark.py,,main_fun,#Any#Any#,6

Before Change

  from tensorflow_estimator.python.estimator.export import export_lib
  export_dir = export_lib.get_timestamped_export_dir(args.export_dir)
  multi_worker_model.save(export_dir, save_format="tf")

  // terminating feed tells spark to skip processing further partitions
  tf_feed.terminate()

After Change

  from tensorflow_estimator.python.estimator.export import export_lib
  export_dir = export_lib.get_timestamped_export_dir(args.export_dir)
  compat.export_saved_model(multi_worker_model, export_dir, ctx.job_name == "chief")

  // terminating feed tells spark to skip processing further partitions
  tf_feed.terminate()

Instances