You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@systemml.apache.org by mb...@apache.org on 2018/05/18 00:06:51 UTC
systemml git commit: [MINOR] Fix keras2dml tabs/spaces inconsistency
Repository: systemml
Updated Branches:
refs/heads/master c93d80602 -> 46f4d9207
[MINOR] Fix keras2dml tabs/spaces inconsistency
Prevents conflict with some IDEs causing errors when mixing tabs and spaces
Closes #763.
Project: http://git-wip-us.apache.org/repos/asf/systemml/repo
Commit: http://git-wip-us.apache.org/repos/asf/systemml/commit/46f4d920
Tree: http://git-wip-us.apache.org/repos/asf/systemml/tree/46f4d920
Diff: http://git-wip-us.apache.org/repos/asf/systemml/diff/46f4d920
Branch: refs/heads/master
Commit: 46f4d920775cdef595a65bce4a134fe256cce28f
Parents: c93d806
Author: Jeff Macaluso <Je...@users.noreply.github.com>
Authored: Thu May 17 16:55:52 2018 -0700
Committer: Matthias Boehm <mb...@gmail.com>
Committed: Thu May 17 16:55:52 2018 -0700
----------------------------------------------------------------------
src/main/python/systemml/mllearn/estimators.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/systemml/blob/46f4d920/src/main/python/systemml/mllearn/estimators.py
----------------------------------------------------------------------
diff --git a/src/main/python/systemml/mllearn/estimators.py b/src/main/python/systemml/mllearn/estimators.py
index de8aeb9..52fe9cc 100644
--- a/src/main/python/systemml/mllearn/estimators.py
+++ b/src/main/python/systemml/mllearn/estimators.py
@@ -946,7 +946,7 @@ class Keras2DML(Caffe2DML):
self.name = keras_model.name
createJavaObject(sparkSession._sc, 'dummy')
if not hasattr(keras_model, 'optimizer'):
- keras_model.compile(loss='categorical_crossentropy', optimizer=keras.optimizers.SGD(lr=0.01, momentum=0.95, decay=5e-4, nesterov=True))
+ keras_model.compile(loss='categorical_crossentropy', optimizer=keras.optimizers.SGD(lr=0.01, momentum=0.95, decay=5e-4, nesterov=True))
convertKerasToCaffeNetwork(keras_model, self.name + ".proto", int(batch_size))
convertKerasToCaffeSolver(keras_model, self.name + ".proto", self.name + "_solver.proto", int(max_iter), int(test_iter), int(test_interval), int(display), lr_policy, weight_decay, regularization_type)
self.weights = tempfile.mkdtemp() if weights is None else weights