baseline_pytorch.json

{
  "model_id": "KBLab/megatron-bert-base-swedish-cased-600k",
  "fill-mask": {
    "input_text": "Det svenska ordet for kung ar [MASK].",
    "mask_index": 7,
    "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.",
    "hidden_shape": [
      1,
      18,
      768
    ],
    "cls_embedding_norm": 8.358345985412598
  }
}