Package io.github.kirstenali.deepj.transformer

Class DeepJPrismTransformerBuilder

java.lang.Object

io.github.kirstenali.deepj.transformer.DeepJPrismTransformerBuilder

public final class DeepJPrismTransformerBuilder
extends Object

Constructor Summary

Constructors

Constructor	Description
DeepJPrismTransformerBuilder()

Method Summary

Modifier and Type	Method	Description
TransformerStack	build()
DeepJPrismTransformerBuilder	dFF(int dFF)
DeepJPrismTransformerBuilder	dModel(int dModel)
DeepJPrismTransformerBuilder	kvRank(int kvRank)
DeepJPrismTransformerBuilder	maxSeqLen(int maxSeqLen)
DeepJPrismTransformerBuilder	nHeads(int nHeads)
DeepJPrismTransformerBuilder	nLayers(int nLayers)
DeepJPrismTransformerBuilder	parameters(ParameterFactory factory)
DeepJPrismTransformerBuilder	qRank(int qRank)
DeepJPrismTransformerBuilder	random(Random rnd)
DeepJPrismTransformerBuilder	seed(long seed)

Methods inherited from class java.lang.Object

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

Constructor Details

DeepJPrismTransformerBuilder

public DeepJPrismTransformerBuilder()

Method Details

parameters

public DeepJPrismTransformerBuilder parameters(ParameterFactory factory)

dModel

public DeepJPrismTransformerBuilder dModel(int dModel)

nHeads

public DeepJPrismTransformerBuilder nHeads(int nHeads)

dFF

public DeepJPrismTransformerBuilder dFF(int dFF)

nLayers

public DeepJPrismTransformerBuilder nLayers(int nLayers)

maxSeqLen

public DeepJPrismTransformerBuilder maxSeqLen(int maxSeqLen)

qRank

public DeepJPrismTransformerBuilder qRank(int qRank)

kvRank

public DeepJPrismTransformerBuilder kvRank(int kvRank)

seed

public DeepJPrismTransformerBuilder seed(long seed)

random

public DeepJPrismTransformerBuilder random(Random rnd)

build

public TransformerStack build()