File size: 2,170 Bytes
20c906a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
{
  "model_name": "bert-base-multilingual-cased",
  "model_id": "M4_4.1_mBERT",
  "member": 4,
  "spec_compliance": {
    "seed": 42,
    "max_length": 128,
    "batch_size": 16,
    "learning_rate": 2e-05,
    "epochs_trained": 3,
    "total_epochs_limit": 5,
    "early_stopping_patience": 5,
    "primary_metric": "val_macro_f1"
  },
  "training_results": {
    "best_epoch": 3,
    "best_val_macro_f1": 0.6795928116428641,
    "best_val_macro_precision": 0.6946485613019967,
    "best_val_macro_recall": 0.6802384049141252,
    "best_val_weighted_f1": 0.7767666885575343,
    "best_val_accuracy": 0.7700119474313023,
    "training_loss": 0.6579384510566588
  },
  "test_results": {
    "test_loss": 0.925308403400838,
    "test_accuracy": 0.7541866028708134,
    "test_macro_precision": 0.6496175553843826,
    "test_macro_recall": 0.6441460137051936,
    "test_macro_f1": 0.6447829635587705,
    "test_weighted_precision": 0.7632761926177464,
    "test_weighted_recall": 0.7541866028708134,
    "test_weighted_f1": 0.7569916810574127,
    "per_class_f1": {
      "F": 0.8371837183718371,
      "R": 0.7635294117647059,
      "N": 0.8100526008182349,
      "M": 0.6145675265553869,
      "S": 0.19858156028368795
    }
  },
  "hardware": {
    "device": "cuda",
    "cuda_available": true,
    "python_version": "3.11.11",
    "platform": "Linux-6.8.0-87-generic-x86_64-with-glibc2.35",
    "cpu_count": 128,
    "ram_gb": 754.5698852539062,
    "gpu_device": "NVIDIA H100 NVL MIG 1g.24gb",
    "gpu_memory_gb": 23.219666944
  },
  "training_time": {
    "total_seconds": 1386.469988,
    "total_minutes": 23.107833133333333,
    "average_per_epoch": 277.29253600000004,
    "timestamp_start": "2025-12-25T10:13:31.474388",
    "timestamp_end": "2025-12-25T10:36:58.037200"
  },
  "library_versions": {
    "pytorch": "2.7.1+cu118",
    "transformers": "4.57.3",
    "numpy": "1.26.4",
    "pandas": "2.2.3",
    "scikit-learn": "1.6.1",
    "python": "3.11.11"
  },
  "data": {
    "train_samples": 15699,
    "val_samples": 3348,
    "test_samples": 3344,
    "num_classes": 5,
    "class_names": [
      "F",
      "R",
      "N",
      "M",
      "S"
    ]
  }
}