SupervisedModel

case class SupervisedModel[I, M <: GenericModule[I, Variable]](module: M & GenericModule[I, Variable], lossFunction: LossFunction, lossCalculation: LossCalculation[I], printMemoryAllocations: Boolean)(implicit tm: TrainingMode[M])
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any

Value members

Concrete methods

def addTotalLossAndReturnGradientsAndNumExamples(samples: I, target: STen, acc: STen, zeroGrad: Boolean): (Long, Seq[Option[STen]])
def addTotalLossAndReturnNumExamples(samples: I, target: STen, acc: STen): Long
def zeroGrad(): Unit
def zipOptimizer(optimizerFactory: Seq[(STen, PTag)] => Optimizer): ModelWithOptimizer[I, M]

Inherited methods

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