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

LossCalculation

lamp.nn.LossCalculation
trait LossCalculation[I]

Loss and Gradient calculation

Takes samples, target, module, loss function and computes the loss and the gradients

Attributes

Graph
Supertypes
class Object
trait Matchable
class Any
Known subtypes

Members list

Value members

Abstract methods

def apply[M <: GenericModule[I, Variable]](samples: I, target: STen, module: M & GenericModule[I, Variable], lossFunction: LossFunction, computeGradients: Boolean, zeroGradBeforeComputingGradients: Boolean)(implicit scope: Scope): (Variable, Long, Option[Seq[Option[STen]]])
Generated with