Transformer constructor

Transformer({
  1. required int vocabSize,
  2. int dModel = 64,
  3. int heads = 4,
  4. required List<int> headLayers,
  5. int? seed,
})

Implementation

Transformer({
  required this.vocabSize,
  this.dModel = 64,
  this.heads = 4,
  required List<int> headLayers,
  int? seed,
}) : head = ANN(layers: headLayers, seed: seed) {
  if (vocabSize <= 0) throw ArgumentError('vocabSize > 0');
  if (dModel <= 0) throw ArgumentError('dModel > 0');
  if (headLayers.isEmpty) throw ArgumentError('headLayers required');
}