File size: 1,521 Bytes
1d300e4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
 "recorded_with": {
  "contract": "huggingface-mic/bart-base-mic",
  "contract_name": "BART",
  "contract_revision": "6794de58ef70196b251907fec7af326947038422",
  "framework": "transformers",
  "variant": "bfloat16",
  "target": "bart-base",
  "kind": "checkpoint",
  "revision": "aadd2ab0ae0c8268c7c9693540e9904811f36177",
  "environment": {
   "python": "3.11.18",
   "transformers": "5.22.0.dev0",
   "torch": "2.14.0+cu130",
   "torchvision": "0.29.2",
   "torchaudio": "2.11.0+cu130",
   "tokenizers": "0.23.3",
   "numpy": "2.4.6",
   "pillow": "13.0.0",
   "accelerate": "1.16.0.dev0",
   "safetensors": "0.8.0",
   "huggingface-hub": "2.3.0",
   "sentencepiece": "0.2.2",
   "protobuf": "7.36.2",
   "librosa": "0.11.0",
   "soundfile": "0.14.0",
   "scipy": "1.17.1"
  },
  "seconds": 7.91,
  "environment_differs": {
   "torchvision": "0.29.2 (pinned 0.29.1)",
   "numpy": "2.4.6 (pinned 2.3.5)",
   "pillow": "13.0.0 (pinned 12.3.0)",
   "accelerate": "1.16.0.dev0 (pinned 1.15.0)"
  },
  "platform": "cuda"
 },
 "outputs": {
  "runtime": {
   "device": "cuda",
   "dtype": "bfloat16",
   "attn_implementation": "sdpa"
  },
  "encoder_last_hidden_shape": [
   2,
   12,
   768
  ],
  "decoder_logits_probe": [
   33.5,
   6.46875,
   16.25,
   6.25,
   10.8125
  ],
  "training": {
   "loss": 0.3203125,
   "gradients": {
    "model.encoder.layers.0.self_attn.q_proj.weight": true,
    "model.decoder.layers.0.self_attn.q_proj.weight": true,
    "model.shared.weight": true
   },
   "updated": true
  }
 }
}