fromMap static method

Transformer fromMap(
  1. Map<String, dynamic> m, {
  2. int? seed,
})

Implementation

static Transformer fromMap(Map<String, dynamic> m, {int? seed}) {
  final h = ANN.fromMap(m['head'] as Map<String, dynamic>, seed: seed);
  final model = Transformer(
    vocabSize: m['vocabSize'] as int,
    dModel: m['dModel'] as int,
    heads: m['heads'] as int,
    headLayers: h.layers,
    seed: seed,
  );
  model.head.applyParamsFrom(h);
  return model;
}