diff --git a/python/dllib/src/bigdl/dllib/examples/keras/README.md b/python/dllib/src/bigdl/dllib/examples/keras/README.md index af4ece12bc1..a27886f3409 100644 --- a/python/dllib/src/bigdl/dllib/examples/keras/README.md +++ b/python/dllib/src/bigdl/dllib/examples/keras/README.md @@ -50,4 +50,4 @@ ${SPARK_HOME}/bin/spark-submit \ ``` * ```--batchSize``` an option that can be used to set batch size. * ```--max_epoch``` an option that can be used to set how many epochs for which the model is to be trained. -* ```--optimizerVersion``` an option that can be used to set DistriOptimizer version, the default value is "optimizerV2". \ No newline at end of file +* ```--optimizerVersion``` an option that can be used to set DistriOptimizer version, the default value is "optimizerV1". diff --git a/python/dllib/src/bigdl/dllib/examples/keras/imdb_cnn_lstm.py b/python/dllib/src/bigdl/dllib/examples/keras/imdb_cnn_lstm.py index 9cba23b1805..9a7b91f21f6 100644 --- a/python/dllib/src/bigdl/dllib/examples/keras/imdb_cnn_lstm.py +++ b/python/dllib/src/bigdl/dllib/examples/keras/imdb_cnn_lstm.py @@ -65,7 +65,7 @@ def build_keras_model(): parser = OptionParser() parser.add_option("-b", "--batchSize", type=int, dest="batchSize", default="32") parser.add_option("-m", "--max_epoch", type=int, dest="max_epoch", default="2") - parser.add_option("--optimizerVersion", dest="optimizerVersion", default="optimizerV2") + parser.add_option("--optimizerVersion", dest="optimizerVersion", default="optimizerV1") (options, args) = parser.parse_args(sys.argv) keras_model = build_keras_model() @@ -103,4 +103,4 @@ def build_keras_model(): trigger=EveryEpoch(), val_method=[Top1Accuracy()] ) - optimizer.optimize() \ No newline at end of file + optimizer.optimize() diff --git a/python/dllib/src/bigdl/dllib/examples/keras/mnist_cnn.py b/python/dllib/src/bigdl/dllib/examples/keras/mnist_cnn.py index 3bf787c6fe4..309116c588f 100644 --- a/python/dllib/src/bigdl/dllib/examples/keras/mnist_cnn.py +++ b/python/dllib/src/bigdl/dllib/examples/keras/mnist_cnn.py @@ -78,7 +78,7 @@ def build_keras_model(): parser.add_option("-b", "--batchSize", type=int, dest="batchSize", default="128") parser.add_option("-m", "--max_epoch", type=int, dest="max_epoch", default="12") parser.add_option("-d", "--dataPath", dest="dataPath", default="/tmp/mnist") - parser.add_option("--optimizerVersion", dest="optimizerVersion", default="optimizerV2") + parser.add_option("--optimizerVersion", dest="optimizerVersion", default="optimizerV1") (options, args) = parser.parse_args(sys.argv) keras_model = build_keras_model() diff --git a/python/dllib/src/bigdl/dllib/models/lenet/lenet5.py b/python/dllib/src/bigdl/dllib/models/lenet/lenet5.py index 6a1e0fec765..cb4a940d5ed 100644 --- a/python/dllib/src/bigdl/dllib/models/lenet/lenet5.py +++ b/python/dllib/src/bigdl/dllib/models/lenet/lenet5.py @@ -60,7 +60,7 @@ def build_model(class_num): parser.add_option("-t", "--endTriggerType", dest="endTriggerType", default="epoch") parser.add_option("-n", "--endTriggerNum", type=int, dest="endTriggerNum", default="20") parser.add_option("-d", "--dataPath", dest="dataPath", default="/tmp/mnist") - parser.add_option("--optimizerVersion", dest="optimizerVersion", default="optimizerV2") + parser.add_option("--optimizerVersion", dest="optimizerVersion", default="optimizerV1") (options, args) = parser.parse_args(sys.argv) diff --git a/python/dllib/src/bigdl/dllib/models/textclassifier/textclassifier.py b/python/dllib/src/bigdl/dllib/models/textclassifier/textclassifier.py index dfa731f6680..c4d0c5d6e12 100644 --- a/python/dllib/src/bigdl/dllib/models/textclassifier/textclassifier.py +++ b/python/dllib/src/bigdl/dllib/models/textclassifier/textclassifier.py @@ -164,7 +164,7 @@ def train(sc, data_path, parser.add_option("--model", dest="model_type", default="cnn") parser.add_option("-p", "--p", dest="p", default="0.0") parser.add_option("-d", "--data_path", dest="data_path", default="/tmp/news20/") - parser.add_option("--optimizerVersion", dest="optimizerVersion", default="optimizerV2") + parser.add_option("--optimizerVersion", dest="optimizerVersion", default="optimizerV1") (options, args) = parser.parse_args(sys.argv) if options.action == "train":