TextTransformer constructor

TextTransformer({
  1. required int vocabSize,
  2. required int maxSeqLen,
  3. required int embedSize,
  4. int numLayers = 4,
  5. int numHeads = 4,
})

Implementation

TextTransformer({
  required int vocabSize,
  required int maxSeqLen,
  required this.embedSize,
  int numLayers = 4,
  int numHeads = 4,
}) : encoder = TransformerEncoder(
       vocabSize: vocabSize,
       embedSize: embedSize,
       blockSize: maxSeqLen,
       numLayers: numLayers,
       numHeads: numHeads,
     );