Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
import $ivy.`org.nd4j::nd4s:0.8.0`
import $ivy.`com.thoughtworks.deeplearning::plugins-indarrayweights:2.0.0-RC5`
import com.thoughtworks.deeplearning.plugins.INDArrayWeights
trait INDArrayLearningRate extends INDArrayWeights {
val learningRate: Double
trait INDArrayOptimizerApi extends super.INDArrayOptimizerApi { this: INDArrayOptimizer =>
import org.nd4s.Implicits._
override def delta: INDArray = {
super.delta * learningRate
}
}
override type INDArrayOptimizer <: Optimizer with INDArrayOptimizerApi
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.