699508c642a95537853090995a83fad992e6bb51,examples/mnist/keras/mnist_tf.py,,main_fun,#Any#Any#,6

Before Change

    multi_worker_model = build_and_compile_cnn_model()
  multi_worker_model.fit(x=train_datasets, epochs=args.epochs, steps_per_epoch=args.steps_per_epoch, callbacks=callbacks)

  if ctx.job_name == "chief":
    from tensorflow_estimator.python.estimator.export import export_lib
    export_dir = export_lib.get_timestamped_export_dir(args.export_dir)
    tf.keras.experimental.export_saved_model(multi_worker_model, export_dir)
    // multi_worker_model.save(args.model_dir, save_format="tf")


if __name__ == "__main__":
  import argparse
  from pyspark.context import SparkContext
  from pyspark.conf import SparkConf

After Change

  multi_worker_model.fit(x=train_datasets, epochs=args.epochs, steps_per_epoch=args.steps_per_epoch, callbacks=callbacks)

  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")


if __name__ == "__main__":
  import argparse

Instances