Package ai.djl.pytorch.zoo.nlp.qa

Class PtBertQATranslatorFactory

java.lang.Object

ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslatorFactory

All Implemented Interfaces:

ai.djl.translate.TranslatorFactory

public class PtBertQATranslatorFactory
extends Object
implements ai.djl.translate.TranslatorFactory

An TranslatorFactory that creates a PtBertQATranslator instance.

Constructor Summary

Constructors

Constructor	Description
PtBertQATranslatorFactory()

Method Summary

Modifier and Type	Method	Description
Set<ai.djl.util.Pair<Type,Type>>	getSupportedTypes()
<I, O> ai.djl.translate.Translator<I,O>	newInstance(Class<I> input, Class<O> output, ai.djl.Model model, Map<String,?> arguments)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ai.djl.translate.TranslatorFactory

isSupported

Constructor Details

PtBertQATranslatorFactory

public PtBertQATranslatorFactory()

Method Details

getSupportedTypes

public Set<ai.djl.util.Pair<Type,Type>> getSupportedTypes()

Specified by:

getSupportedTypes in interface ai.djl.translate.TranslatorFactory

newInstance

public <I,
O> ai.djl.translate.Translator<I,O> newInstance(Class<I> input,
 Class<O> output,
 ai.djl.Model model,
 Map<String,?> arguments)

Specified by:

newInstance in interface ai.djl.translate.TranslatorFactory