org.apache.mxnet

Optimizer

Related Docs: object Optimizer | package mxnet

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
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new Optimizer()

Abstract Value Members

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

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

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

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

  5. 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

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

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

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

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

    Definition Classes
    Any
  5. def clone(): AnyRef

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

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

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

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

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

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

    Attributes
    protected
  12. def hashCode(): Int

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

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

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

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

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

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

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

    Definition Classes
    AnyRef
  20. var numUpdate: Int

    Attributes
    protected
  21. var rescaleGrad: Float

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

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

  24. 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.

  25. def setRescaleGrad(rescaleGrad: Float): Unit

  26. def setSymbol(sym: Symbol): Unit

  27. 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.

  28. var specialized: Boolean

    Attributes
    protected
  29. var symbol: Symbol

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

    Definition Classes
    AnyRef
  31. def toString(): String

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

    update num_update

    update num_update

    index

    The index will be updated

    Attributes
    protected
  33. final def wait(): Unit

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

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

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

    Attributes
    protected
  37. val weightSet: Set[Int]

    Attributes
    protected

Deprecated Value Members

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

    Annotations
    @deprecated
    Deprecated

    Use setLrMult instead.

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped