model/model_epoch_1_loss_6.5042_metadata.json

{
  "epoch": 1,
  "val_loss": 6.504229426383972,
  "model_config": {
    "vocab_size": 5477,
    "embed_size": 768,
    "num_heads": 12,
    "num_layers": 8,
    "hidden_size": 2048,
    "max_length": 512
  },
  "tokenizer_path": "model_checkpoints/model_epoch_1_loss_6.5042_tokenizer.json"
}