Package com.yahoo.embedding

Class BertBaseEmbedderConfig

java.lang.Object

com.yahoo.config.Node

com.yahoo.config.InnerNode

com.yahoo.config.ConfigInstance

com.yahoo.embedding.BertBaseEmbedderConfig

public final class BertBaseEmbedderConfig
extends com.yahoo.config.ConfigInstance

This class represents the root node of bert-base-embedder Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	BertBaseEmbedderConfig.Builder
static final class	BertBaseEmbedderConfig.OnnxExecutionMode	This class represents bert-base-embedder.onnxExecutionMode Settings for ONNX model evaluation
static final class	BertBaseEmbedderConfig.PoolingStrategy	This class represents bert-base-embedder.poolingStrategy Pooling strategy
static interface	BertBaseEmbedderConfig.Producer

Field Summary

Fields

Modifier and Type	Field	Description
static final String	CONFIG_DEF_MD5
static final String	CONFIG_DEF_NAME
static final String	CONFIG_DEF_NAMESPACE
static final String[]	CONFIG_DEF_SCHEMA

Constructor Summary

Constructors

Constructor	Description
BertBaseEmbedderConfig(BertBaseEmbedderConfig.Builder builder)

Method Summary

Modifier and Type	Method	Description
static String	getDefMd5()
static String	getDefName()
static String	getDefNamespace()
BertBaseEmbedderConfig.OnnxExecutionMode.Enum	onnxExecutionMode()
int	onnxGpuDevice()
int	onnxInterOpThreads()
int	onnxIntraOpThreads()
BertBaseEmbedderConfig.PoolingStrategy.Enum	poolingStrategy()
Path	tokenizerVocab()
com.yahoo.config.ModelReference	tokenizerVocabReference()
String	transformerAttentionMask()
int	transformerEndSequenceToken()
String	transformerInputIds()
int	transformerMaxTokens()
Path	transformerModel()
com.yahoo.config.ModelReference	transformerModelReference()
String	transformerOutput()
int	transformerStartSequenceToken()
String	transformerTokenTypeIds()

Methods inherited from class com.yahoo.config.ConfigInstance

getDefName, getDefNamespace, serialize, serialize

Methods inherited from class com.yahoo.config.InnerNode

equals, getAllDescendantLeafNodes, getChildren, getChildrenWithVectorsFlattened, hashCode, postInitialize, toString

Methods inherited from class com.yahoo.config.Node

clone

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, wait, wait, wait

Field Details

CONFIG_DEF_MD5

public static final String CONFIG_DEF_MD5

See Also:

• Constant Field Values

CONFIG_DEF_NAME

public static final String CONFIG_DEF_NAME

See Also:

• Constant Field Values

CONFIG_DEF_NAMESPACE

public static final String CONFIG_DEF_NAMESPACE

See Also:

• Constant Field Values

CONFIG_DEF_SCHEMA

public static final String[] CONFIG_DEF_SCHEMA

Constructor Details

BertBaseEmbedderConfig

public BertBaseEmbedderConfig(BertBaseEmbedderConfig.Builder builder)

Method Details

getDefMd5

public static String getDefMd5()

getDefName

public static String getDefName()

getDefNamespace

public static String getDefNamespace()

tokenizerVocab

public Path tokenizerVocab()

Returns:

bert-base-embedder.tokenizerVocab

tokenizerVocabReference

public com.yahoo.config.ModelReference tokenizerVocabReference()

Returns:

bert-base-embedder.tokenizerVocab ModelReference

transformerModel

public Path transformerModel()

Returns:

bert-base-embedder.transformerModel

transformerModelReference

public com.yahoo.config.ModelReference transformerModelReference()

Returns:

bert-base-embedder.transformerModel ModelReference

transformerMaxTokens

public int transformerMaxTokens()

Returns:

bert-base-embedder.transformerMaxTokens

poolingStrategy

public BertBaseEmbedderConfig.PoolingStrategy.Enum poolingStrategy()

Returns:

bert-base-embedder.poolingStrategy

transformerInputIds

public String transformerInputIds()

Returns:

bert-base-embedder.transformerInputIds

transformerAttentionMask

public String transformerAttentionMask()

Returns:

bert-base-embedder.transformerAttentionMask

transformerTokenTypeIds

public String transformerTokenTypeIds()

Returns:

bert-base-embedder.transformerTokenTypeIds

transformerStartSequenceToken

public int transformerStartSequenceToken()

Returns:

bert-base-embedder.transformerStartSequenceToken

transformerEndSequenceToken

public int transformerEndSequenceToken()

Returns:

bert-base-embedder.transformerEndSequenceToken

transformerOutput

public String transformerOutput()

Returns:

bert-base-embedder.transformerOutput

onnxExecutionMode

public BertBaseEmbedderConfig.OnnxExecutionMode.Enum onnxExecutionMode()

Returns:

bert-base-embedder.onnxExecutionMode

onnxInterOpThreads

public int onnxInterOpThreads()

Returns:

bert-base-embedder.onnxInterOpThreads

onnxIntraOpThreads

public int onnxIntraOpThreads()

Returns:

bert-base-embedder.onnxIntraOpThreads

onnxGpuDevice

public int onnxGpuDevice()

Returns:

bert-base-embedder.onnxGpuDevice