Optimizer

Abstract Value Members

abstract def createState(index: Int, weight: NDArray): AnyRef
abstract def deserializeState(bytes: Array[Byte]): AnyRef
abstract def disposeState(state: AnyRef): Unit
abstract def serializeState(state: AnyRef): Array[Byte]
abstract def update(index: Int, weight: NDArray, grad: NDArray, state: AnyRef): Unit

Update the parameters.
Update the parameters.
index
An unique integer key used to index the parameters
weight
weight ndarray
grad
grad ndarray
state
NDArray or other objects returned by initState The auxiliary state used in optimization.

Concrete Value Members

final def !=(arg0: Any): Boolean

Definition Classes
AnyRef → Any
final def ##(): Int

Definition Classes
AnyRef → Any
final def ==(arg0: Any): Boolean

Definition Classes
AnyRef → Any
final def asInstanceOf[T0]: T0

Definition Classes
Any
def clone(): AnyRef

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( ... )
final def eq(arg0: AnyRef): Boolean

Definition Classes
AnyRef
def equals(arg0: Any): Boolean

Definition Classes
AnyRef → Any
def finalize(): Unit

Attributes
protected[java.lang]
Definition Classes
AnyRef
Annotations
@throws( classOf[java.lang.Throwable] )
final def getClass(): Class[_]

Definition Classes
AnyRef → Any
def getLr(index: Int, lr: Float): Float

Attributes
protected
def getWd(index: Int, wd: Float): Float

Attributes
protected
def hashCode(): Int

Definition Classes
AnyRef → Any
var idx2name: Map[Int, String]

Attributes
protected
val indexUpdateCount: Map[Int, Int]

Attributes
protected
final def isInstanceOf[T0]: Boolean

Definition Classes
Any
val lrMult: Map[Either[Int, String], Float]

Attributes
protected
final def ne(arg0: AnyRef): Boolean

Definition Classes
AnyRef
final def notify(): Unit

Definition Classes
AnyRef
final def notifyAll(): Unit

Definition Classes
AnyRef
var numUpdate: Int

Attributes
protected
var rescaleGrad: Float

Attributes
protected
def setArgNames(argNames: Seq[String]): Unit
def setIdx2Name(paramIdx2Name: Map[Int, String]): Unit
def setLrMult(argsLrMult: Map[Either[Int, String], Float]): Unit

Sets an individual learning rate multiplier for each parameter.
Sets an individual learning rate multiplier for each parameter. If you specify a learning rate multiplier for a parameter, then the learning rate for the parameter will be set as the product of the global learning rate and its multiplier. note:: The default learning rate multiplier of a Variable can be set with lr_mult argument in the constructor.
def setRescaleGrad(rescaleGrad: Float): Unit
def setSymbol(sym: Symbol): Unit
def setWdMult(argsWdMult: Map[Either[Int, String], Float]): Unit

Sets an individual weight decay multiplier for each parameter.
Sets an individual weight decay multiplier for each parameter.
By default, the weight decay multipler is set as 0 for all parameters whose name don't end with _weight or _gamma, if you call the setIdx2Name method to set idx2name.
note:: The default weight decay multiplier for a Variable can be set with its wd_mult argument in the constructor.
var specialized: Boolean

Attributes
protected
var symbol: Symbol

Attributes
protected
final def synchronized[T0](arg0: ⇒ T0): T0

Definition Classes
AnyRef
def toString(): String

Definition Classes
AnyRef → Any
def updateCount(index: Int): Unit

update num_update
update num_update
index
The index will be updated

Attributes
protected
final def wait(): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long, arg1: Int): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
final def wait(arg0: Long): Unit

Definition Classes
AnyRef
Annotations
@throws( ... )
val wdMult: Map[Either[Int, String], Float]

Attributes
protected
val weightSet: Set[Int]

Attributes
protected

Deprecated Value Members

def setLrScale(lrScale: Map[Int, Float]): Unit

Annotations
@deprecated
Deprecated
Use setLrMult instead.

Related Docs: object Optimizer | package mxnet

abstract class Optimizer extends Serializable

Instance Constructors

new Optimizer()

Abstract Value Members

abstract def createState(index: Int, weight: NDArray): AnyRef

abstract def deserializeState(bytes: Array[Byte]): AnyRef

abstract def disposeState(state: AnyRef): Unit

abstract def serializeState(state: AnyRef): Array[Byte]

abstract def update(index: Int, weight: NDArray, grad: NDArray, state: AnyRef): Unit

Concrete Value Members

final def !=(arg0: Any): Boolean

final def ##(): Int

final def ==(arg0: Any): Boolean

final def asInstanceOf[T0]: T0

def clone(): AnyRef

final def eq(arg0: AnyRef): Boolean

def equals(arg0: Any): Boolean

def finalize(): Unit

final def getClass(): Class[_]

def getLr(index: Int, lr: Float): Float

def getWd(index: Int, wd: Float): Float

def hashCode(): Int

var idx2name: Map[Int, String]

val indexUpdateCount: Map[Int, Int]

final def isInstanceOf[T0]: Boolean

val lrMult: Map[Either[Int, String], Float]

final def ne(arg0: AnyRef): Boolean

final def notify(): Unit

final def notifyAll(): Unit

var numUpdate: Int

var rescaleGrad: Float

def setArgNames(argNames: Seq[String]): Unit

def setIdx2Name(paramIdx2Name: Map[Int, String]): Unit

def setLrMult(argsLrMult: Map[Either[Int, String], Float]): Unit

def setRescaleGrad(rescaleGrad: Float): Unit

def setSymbol(sym: Symbol): Unit

def setWdMult(argsWdMult: Map[Either[Int, String], Float]): Unit

var specialized: Boolean

var symbol: Symbol

final def synchronized[T0](arg0: ⇒ T0): T0

def toString(): String

def updateCount(index: Int): Unit

final def wait(): Unit

final def wait(arg0: Long, arg1: Int): Unit

final def wait(arg0: Long): Unit

val wdMult: Map[Either[Int, String], Float]

val weightSet: Set[Int]

Deprecated Value Members

def setLrScale(lrScale: Map[Int, Float]): Unit

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped