Packages

case class AdamW(parameters: Seq[(STen, PTag)], weightDecay: OptimizerHyperparameter, learningRate: OptimizerHyperparameter = simple(0.001), beta1: OptimizerHyperparameter = simple(0.9), beta2: OptimizerHyperparameter = simple(0.999), eps: Double = 1e-8, clip0: Option[Double] = None, debias: Boolean = true, mixedPrecision: Boolean = false) extends Optimizer with Product with Serializable

See also

https://arxiv.org/pdf/1711.05101.pdf Algorithm 2

Linear Supertypes
Serializable, Product, Equals, Optimizer, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. AdamW
  2. Serializable
  3. Product
  4. Equals
  5. Optimizer
  6. AnyRef
  7. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. Protected

Instance Constructors

  1. new AdamW(parameters: Seq[(STen, PTag)], weightDecay: OptimizerHyperparameter, learningRate: OptimizerHyperparameter = simple(0.001), beta1: OptimizerHyperparameter = simple(0.9), beta2: OptimizerHyperparameter = simple(0.999), eps: Double = 1e-8, clip0: Option[Double] = None, debias: Boolean = true, mixedPrecision: Boolean = false)

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. val beta1: OptimizerHyperparameter
  6. val beta2: OptimizerHyperparameter
  7. val clip: Option[STen]
  8. val clip0: Option[Double]
  9. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.CloneNotSupportedException]) @IntrinsicCandidate() @native()
  10. val debias: Boolean
  11. def downCast(t: STen, target: STen)(implicit scope: Scope): STen
  12. val eps: Double
  13. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  14. final def getClass(): Class[_ <: AnyRef]
    Definition Classes
    AnyRef → Any
    Annotations
    @IntrinsicCandidate() @native()
  15. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  16. val learningRate: OptimizerHyperparameter
  17. def load(tensors: Seq[STen]): Unit
    Definition Classes
    AdamWOptimizer
  18. val mixedPrecision: Boolean
  19. val mt: List[STen]
  20. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  21. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @IntrinsicCandidate() @native()
  22. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @IntrinsicCandidate() @native()
  23. val parameters: Seq[(STen, PTag)]
  24. def productElementNames: Iterator[String]
    Definition Classes
    Product
  25. def release(): Unit
    Definition Classes
    AdamWOptimizer
  26. val scope0: Scope
  27. def state: List[STen]
    Definition Classes
    AdamWOptimizer
  28. def step(gradients: Seq[Option[STen]], scheduleFactor: Double): Unit
    Definition Classes
    AdamWOptimizer
  29. var stepCount: Long
  30. val stepCountSTen: STen
  31. final def synchronized[T0](arg0: => T0): T0
    Definition Classes
    AnyRef
  32. def upCast(t: STen)(implicit scope: Scope): STen
  33. val vt: List[STen]
  34. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  35. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException]) @native()
  36. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  37. val weightDecay: OptimizerHyperparameter
  38. val workingCopy: List[Option[STen]]

Deprecated Value Members

  1. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.Throwable]) @Deprecated
    Deprecated

    (Since version 9)

Inherited from Serializable

Inherited from Product

Inherited from Equals

Inherited from Optimizer

Inherited from AnyRef

Inherited from Any

Ungrouped