training
Experimental entry point for the training DSL.
Example:
val runner = training<FP32, Float> {
model { myModel }
loss { crossEntropyLoss() }
optimizer { sgd(lr = 0.01) }
}
runner.step(ctx, x, y)Content copied to clipboard
Experimental entry point for the training DSL.
Example:
val runner = training<FP32, Float> {
model { myModel }
loss { crossEntropyLoss() }
optimizer { sgd(lr = 0.01) }
}
runner.step(ctx, x, y)