Package examples.inference.saveload

Types

SaveTrainedModelHelper

class SaveTrainedModelHelper(trainBatchSize: Int, testBatchSize: Int)

The object wraps the logic on a given model training up to particular accuracy on a test dataset and then persist it in a file.

eluLenetOnMnistWithIntermediateSave

This examples demonstrates running Save and Load for prediction on mnist dataset.

main

fun main()

reluLenetOnMnistWithIntermediateSave

This examples demonstrates running Save and Load for prediction on mnist dataset.