loadFromMap method

void loadFromMap(Map<String, Tensor> weights)

Load weights from an in-memory map.

weights: map of parameter name → Tensor

Implementation

void loadFromMap(Map<String, Tensor> weights) {
  for (final entry in weights.entries) {
    try {
      _loadSingleWeight(entry.key, entry.value);
    } catch (_) {
      // Silently skip weights that don't match
    }
  }
}