ai.djl.pytorch.engine

Class PtPredictor<I,O>

java.lang.Object

ai.djl.inference.BasePredictor<I,O>

ai.djl.pytorch.engine.PtPredictor<I,O>

Type Parameters:

I - the input object

O - the output object

All Implemented Interfaces:

ai.djl.inference.Predictor<I,O>, java.lang.AutoCloseable

public class PtPredictor<I,O>
extends ai.djl.inference.BasePredictor<I,O>

PtPredictor is the Pytorch implementation of Predictor.

See Also:

Predictor

Field Summary

Fields inherited from class ai.djl.inference.BasePredictor

manager, model, prepared

Method Summary

Modifier and Type	Method and Description
protected void	finalize()

Methods inherited from class ai.djl.inference.BasePredictor

batchPredict, close, forward, predict, setMetrics, waitToRead

Methods inherited from class java.lang.Object

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

Method Detail

finalize

protected void finalize()
                 throws java.lang.Throwable

Overrides:

finalize in class java.lang.Object

Throws:

java.lang.Throwable