From a314ebbef0f98722b6bdf06dfd60dc7201ecef93 Mon Sep 17 00:00:00 2001 From: Yina Chen <33650826+cyita@users.noreply.github.com> Date: Wed, 9 Sep 2020 09:49:07 +0800 Subject: [PATCH] Orca tf save_keras_model add overwrite option (#2835) * orca tf save_keras_model add overwrite option * add default value & update docs * fix --- python/orca/example/learn/tf/lenet_mnist_keras.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/orca/example/learn/tf/lenet_mnist_keras.py b/python/orca/example/learn/tf/lenet_mnist_keras.py index 756f01cb921..2e8fe04884c 100644 --- a/python/orca/example/learn/tf/lenet_mnist_keras.py +++ b/python/orca/example/learn/tf/lenet_mnist_keras.py @@ -64,6 +64,7 @@ def main(max_epoch): est.save_keras_model("/tmp/mnist_keras.h5") + if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('--cluster_mode', type=str, default="local",