Package chat.octet.model.components.criteria.impl

Class MaxTimeCriteria

java.lang.Object

chat.octet.model.components.criteria.impl.MaxTimeCriteria

All Implemented Interfaces:

StoppingCriteria

public class MaxTimeCriteria
extends Object
implements StoppingCriteria

Constructor Summary

Constructors

Constructor	Description
MaxTimeCriteria(long maxTimeMillis)
MaxTimeCriteria(long maxTimeMillis, long initialTimestamp)

Method Summary

Modifier and Type	Method	Description
boolean	criteria(int[] inputTokenIds, float[] scores, Object... args)	Stopping criteria

Methods inherited from class java.lang.Object

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

Constructor Details

MaxTimeCriteria

public MaxTimeCriteria(long maxTimeMillis)

MaxTimeCriteria

public MaxTimeCriteria(long maxTimeMillis,
 long initialTimestamp)

Method Details

criteria

public boolean criteria(@Nullable
 int[] inputTokenIds,
 @Nonnull
 float[] scores,
 Object... args)

Description copied from interface: StoppingCriteria

Stopping criteria

Specified by:

criteria in interface StoppingCriteria

Parameters:

inputTokenIds - Indices of input sequence tokens in the vocabulary.

scores - Prediction scores of a language modeling head. These can be logits for each vocabulary.

args - Specific args to a stopping criteria.

Returns:

boolean `False` indicates we should continue, `True` indicates we should stop.