validation_report.json

{
  "model_id": "KBLab/megatron-bert-base-swedish-cased-600k",
  "fill-mask": {
    "shape": [
      1,
      10,
      64128
    ],
    "max_abs_diff": 6.884336471557617e-05,
    "mean_abs_diff": 8.68798088049516e-06,
    "input_text": "Det svenska ordet for kung ar [MASK].",
    "mask_index": 7,
    "pytorch": {
      "logits_shape": [
        1,
        10,
        64128
      ],
      "top5_token_ids": [
        37090,
        57980,
        35856,
        50325,
        55930
      ],
      "top5_tokens": [
        "kung",
        "hertig",
        "fe",
        "bonde",
        "konung"
      ]
    },
    "onnx_fp32": {
      "logits_shape": [
        1,
        10,
        64128
      ],
      "top5_token_ids": [
        37090,
        57980,
        35856,
        50325,
        55930
      ],
      "top5_tokens": [
        "kung",
        "hertig",
        "fe",
        "bonde",
        "konung"
      ]
    }
  },
  "feature-extraction": {
    "input_text": "Svensk BERT ska kunna skapa embeddings for den har texten.",
    "last_hidden_state": {
      "pytorch_shape": [
        1,
        18,
        768
      ],
      "onnx_fp32_shape": [
        1,
        18,
        768
      ],
      "max_abs_diff": 5.930662155151367e-06,
      "mean_abs_diff": 1.0840861932592816e-06
    },
    "cls_embedding": {
      "pytorch_shape": [
        1,
        768
      ],
      "onnx_fp32_shape": [
        1,
        768
      ],
      "max_abs_diff": 2.950429916381836e-06,
      "mean_abs_diff": 5.458733198793198e-07,
      "cosine_similarity": 1.0
    }
  }
}