Skip to content

Commit

Permalink
Orca tf save_keras_model add overwrite option (#2835)
Browse files Browse the repository at this point in the history
* orca tf save_keras_model add overwrite option

* add default value & update docs

* fix
  • Loading branch information
cyita committed Sep 9, 2020
1 parent 73a29e9 commit a314ebb
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions python/orca/example/learn/tf/lenet_mnist_keras.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit a314ebb

Please sign in to comment.