{
  "experiment_name": "TypePredictor",
  "dataset_repo_id": "U4RASD/TypePrediction",
  "dataset_files": {
    "train": "type_predictor_train.jsonl",
    "validation": "type_predictor_val.jsonl",
    "test": "type_predictor_test.jsonl"
  },
  "model_name": "U4RASD/NeoAraBERT",
  "model_repo_id": "U4RASD/TypePredictor",
  "seed": 42,
  "max_length": 512,
  "context_chars": 300,
  "context_candidates": [
    null,
    500,
    300,
    150,
    80,
    30,
    0
  ],
  "num_epochs": 4.0,
  "train_batch_size": 4,
  "eval_batch_size": 8,
  "gradient_accumulation_steps": 4,
  "effective_batch_size": 16,
  "encoder_lr": 1e-05,
  "head_lr": 5e-05,
  "dropout": 0.1,
  "weight_decay": 0.01,
  "warmup_ratio": 0.1,
  "max_grad_norm": 1.0,
  "logging_steps": 50,
  "eval_steps": 500,
  "save_steps": 500,
  "fp16": true,
  "best_metric_name": "validation macro F1",
  "parameter_counts": {
    "total": 248162325,
    "trainable": 248162325,
    "encoder": 248146176,
    "classifier": 16149
  },
  "checkpoint_summary": {
    "best_checkpoint_source": "outputs/TypePredictor/checkpoints/checkpoint-25000",
    "best_checkpoint_preserved": "outputs/TypePredictor/best_checkpoint",
    "best_step": 25000,
    "best_epoch": 3.9682923925552602,
    "best_validation_macro_f1": 0.9587281193044483,
    "latest_checkpoint_source": "outputs/TypePredictor/checkpoints/checkpoint-25196",
    "latest_checkpoint_preserved": "outputs/TypePredictor/latest_checkpoint",
    "latest_step": 25196,
    "latest_epoch": 3.9994047382832654,
    "best_and_latest_same_source": false,
    "root_released_model": "best checkpoint"
  },
  "training_started_at_utc": "2026-07-12T15:30:38+00:00",
  "training_finished_at_utc": "2026-07-12T18:55:01+00:00",
  "training_wall_seconds": 12262.467122793198,
  "versions": {
    "python": "3.12.3 (main, Aug 14 2025, 17:47:21) [GCC 13.3.0]",
    "torch": "2.5.1+cu124",
    "transformers": "4.49.0",
    "huggingface_hub": "0.29.1",
    "scikit_learn": "1.5.2",
    "numpy": "2.0.2",
    "pandas": "2.2.3",
    "platform": "Linux-6.8.0-48-generic-x86_64-with-glibc2.39"
  },
  "hardware": {
    "cuda_available": true,
    "gpu_name": "NVIDIA A40",
    "gpu_vram_gb": 44.431640625
  }
}