lamp-core/lamp.nn/Optimizer

Optimizer

trait Optimizer
class Object
trait Matchable
class Any
class AdamW
class RAdam
class SGDW
class Shampoo
class Yogi

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