lamp-core
API
Generated with
lamp-core/lamp/lamp.nn/Optimizer

Optimizer

lamp.nn.Optimizer
trait Optimizer

Attributes

Graph
Supertypes
class Object
trait Matchable
class Any
Known subtypes
class AdamW
class RAdam
class SGDW
class Shampoo
class Yogi

Members list

Value members

Abstract methods

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