Class/Object

ml.dmlc.mxnet

Optimizer

Related Docs: object Optimizer | package mxnet

Permalink

abstract class Optimizer extends Serializable

Linear Supertypes
Serializable, AnyRef, Any
Known Subclasses
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. Optimizer
  2. Serializable
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new Optimizer()

    Permalink

Abstract Value Members

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

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

    Permalink
  3. abstract def disposeState(state: AnyRef): Unit

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

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

    Permalink

    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

  1. final def !=(arg0: Any): Boolean

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

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

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

    Permalink
    Definition Classes
    Any
  5. def clone(): AnyRef

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

    Permalink
    Definition Classes
    AnyRef
  7. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit

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

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

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

    Permalink
    Attributes
    protected
  12. def hashCode(): Int

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

    Permalink
    Attributes
    protected
  14. val indexUpdateCount: Map[Int, Int]

    Permalink
    Attributes
    protected
  15. final def isInstanceOf[T0]: Boolean

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

    Permalink
    Attributes
    protected
  17. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  18. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  19. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  20. var numUpdate: Int

    Permalink
    Attributes
    protected
  21. var rescaleGrad: Float

    Permalink
    Attributes
    protected
  22. def setArgNames(argNames: Seq[String]): Unit

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

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

    Permalink

    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.

  25. def setRescaleGrad(rescaleGrad: Float): Unit

    Permalink
  26. def setSymbol(sym: Symbol): Unit

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

    Permalink

    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.

  28. var specialized: Boolean

    Permalink
    Attributes
    protected
  29. var symbol: Symbol

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

    Permalink
    Definition Classes
    AnyRef
  31. def toString(): String

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

    Permalink

    update num_update

    update num_update

    index

    The index will be updated

    Attributes
    protected
  33. final def wait(): Unit

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

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

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

    Permalink
    Attributes
    protected
  37. val weightSet: Set[Int]

    Permalink
    Attributes
    protected

Deprecated Value Members

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

    Permalink
    Annotations
    @deprecated
    Deprecated

    Use setLrMult instead.

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped