Skip to content

Commit

Permalink
Merge pull request apache#15 from javelinjs/scala-package-cc
Browse files Browse the repository at this point in the history
Adam bug fixing, clean up .gitignore
  • Loading branch information
terrytangyuan committed Dec 28, 2015
2 parents bed8eaf + d9202e3 commit 2a53b83
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 31 deletions.
33 changes: 5 additions & 28 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -87,37 +87,14 @@ example/notebooks/.ipynb_checkpoints/*


# Scala package
# Jetbrain
mxnet-scala/.idea

# ctags
mxnet-scala/tags

mxnet-scala/*.class
mxnet-scala/*.log

# sbt specific
mxnet-scala/.cache
mxnet-scala/.lib/
mxnet-scala/dist/*
mxnet-scala/target/
mxnet-scala/lib_managed/
mxnet-scala/src_managed/
mxnet-scala/project/boot/
mxnet-scala/project/plugins/project/

#scala target folders
*.idea
*.class
scala-package/*/target/
scala-package/*/*/target/

# IDE specific
*.scala_dependencies
*.worksheet
*.idea
*.iml
#eclipse
.classpath
.project
.settings


*.classpath
*.project
*.settings
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ class Adam(var learningRate: Float = 0.002f, val beta1: Float = 0.9f, val beta2:
this.learningRate
}) * lrScale.getOrElse(index, 1f)

var (mean, variance) = state.asInstanceOf[(NDArray, NDArray)]
val (mean, variance) = state.asInstanceOf[(NDArray, NDArray)]

// increment time only when the first parameters is called
if (timeFirstIndex == None) {
Expand Down Expand Up @@ -81,8 +81,8 @@ class Adam(var learningRate: Float = 0.002f, val beta1: Float = 0.9f, val beta2:
}

weight += -step
mean = meanT
variance = varianceT
mean.set(meanT)
variance.set(varianceT)
}

// Create additional optimizer state: mean, variance
Expand Down

0 comments on commit 2a53b83

Please sign in to comment.