ONNXModelHub

class ONNXModelHub(cacheDirectory: File) : ModelHub

This class provides methods for loading ONNX model to the local cacheDirectory.

Since

0.3

Parameters

cacheDirectory

The directory for all loaded models. It should be created before model loading and should have all required permissions for file writing/reading on your OS.

Constructors

ONNXModelHub

fun ONNXModelHub(cacheDirectory: File)

Functions

get

operator fun <T : InferenceModel, U : InferenceModel> get(modelType: ModelType<T, U>): U

loadModel

open override fun <T : InferenceModel, U : InferenceModel> loadModel(modelType: ModelType<T, U>, loadingMode: LoadingMode): T

Loads model configuration without weights.

loadPretrainedModel

fun <T : InferenceModel, U : InferenceModel> loadPretrainedModel(modelType: ModelType<T, U>, loadingMode: LoadingMode): U

Properties

cacheDirectory

val cacheDirectory: File