MnistMpl

Simple MLP for MNIST as a models-as-code example.

Constructors

constructor()

open suspend override fun calculate(module: Module<FP32, Float>, inputValue: Tensor<FP32, Float>, executionContext: ExecutionContext, reportProgress: suspend (current: Int, total: Int, message: String?) -> Unit): Tensor<FP32, Float>

Implement this in your models. Call reportProgress whenever you want to emit a progress update.

open override fun create(executionContext: ExecutionContext): Module<FP32, Float>

open override fun modelCard(): ModelCard