loadWeightsFromMap method

void loadWeightsFromMap(
  1. Map<String, Tensor> weights
)

Load weights from a map of tensor name → Tensor.

Implementation

void loadWeightsFromMap(Map<String, Tensor> weights) {
  final loader = DonutWeightLoader(
    encoder: encoder,
    decoder: decoder,
  );
  loader.loadFromMap(weights);
  _isLoaded = true;
}