toMap method

Map<String, dynamic> toMap()

Implementation

Map<String, dynamic> toMap() => {
  'vocabSize': vocabSize,
  'dModel': dModel,
  'heads': heads,
  'head': head.toMap(),
};