lamp-core/lamp.nn/SGDW

SGDW

case class SGDW(parameters: Seq[(STen, PTag)], learningRate: OptimizerHyperparameter, weightDecay: OptimizerHyperparameter, momentum: Option[OptimizerHyperparameter], clip: Option[Double]) extends Optimizer
Companion:
object
trait Serializable
trait Product
trait Equals
trait Optimizer
class Object
trait Matchable
class Any

Value members

Concrete methods

def load(tensors: Seq[STen]): Unit
def release(): Unit
def state: Seq[STen]
def step(gradients: Seq[Option[STen]], scheduleFactor: Double): Unit

Inherited methods

def productElementNames: Iterator[String]
Inherited from:
Product
def productIterator: Iterator[Any]
Inherited from:
Product

Concrete fields

val scope: Scope
val velocity: Seq[Option[(STen, OptimizerHyperparameter)]]