SimpleLossCalculation

lamp.nn.SimpleLossCalculation

Attributes

Graph
Supertypes
trait LossCalculation[I]
class Object
trait Matchable
class Any

Members list

Value members

Concrete 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]]])