skainet-lang-core/sk.ainet.java/TrainingLoop/Builder

Builder

class Builder(source)

Builder for TrainingLoop.

Constructors

constructor()

Functions

fun build(): TrainingLoop

fun context(ctx: ExecutionContext): TrainingLoop.Builder

fun loss(loss: Loss): TrainingLoop.Builder

fun model(model: Module<*, *>): TrainingLoop.Builder

fun optimizer(optimizer: Optimizer): TrainingLoop.Builder