loadPretrainedModel

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

Loads pretrained model of modelType from the ModelHub in loadingMode.

Return

Pretrained model.

Parameters

modelType

This unique identifier defines the way to the S3 bucket with the model and its weights and the local directory for the model and its weights.

loadingMode

Strategy of existing model use-case handling.