Hugging Face's logo

augmem
/
AS-20M

Feature Extraction
PyTorch
English
native_efficientat_audio_embedding
audio
speech
embedding
retrieval
efficientat
matryoshka
memory-augmentation
Model card Files
xet
 Community
AS-20M
File size: 3,200 Bytes
7a716db
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
80
{
  "artifacts": {
    "safetensors": {
      "bytes": 79578032,
      "parameter_count": 19837720,
      "path": "AS-20M.safetensors",
      "sha256": "77152ea3cd4a9f841cb88230b829cce3fe68afa3d1eef7b41e01f32537859ca3",
      "state_tensor_count": 312,
      "state_tensor_elements": 19886566
    }
  },
  "canonical_name_rule": "<modality>-<size>, modalities sorted alphabetically",
  "checkpoint_epoch": 4,
  "checkpoint_metrics": {
    "audio_cos": 0.8108276128768921,
    "embed_pearson": 0.7953315377235413,
    "sim_pearson": 0.88530433177948,
    "student_at_r10_128": 0.6377999782562256,
    "student_at_r10_1280": 0.6571999788284302,
    "student_at_r10_256": 0.6527999639511108,
    "student_at_r10_512": 0.6563999652862549,
    "student_at_r10_768": 0.6570000052452087,
    "student_at_r1_128": 0.29739999771118164,
    "student_at_r1_1280": 0.3215999901294708,
    "student_at_r1_256": 0.31439998745918274,
    "student_at_r1_512": 0.3203999996185303,
    "student_at_r1_768": 0.3215999901294708,
    "student_at_r5_128": 0.5307999849319458,
    "student_at_r5_1280": 0.5541999936103821,
    "student_at_r5_256": 0.550000011920929,
    "student_at_r5_512": 0.5527999997138977,
    "student_at_r5_768": 0.5533999800682068,
    "student_ta_r10_128": 0.649399995803833,
    "student_ta_r10_1280": 0.6651999950408936,
    "student_ta_r10_256": 0.6615999937057495,
    "student_ta_r10_512": 0.6625999808311462,
    "student_ta_r10_768": 0.663599967956543,
    "student_ta_r1_128": 0.2793999910354614,
    "student_ta_r1_1280": 0.3027999997138977,
    "student_ta_r1_256": 0.29919999837875366,
    "student_ta_r1_512": 0.30140000581741333,
    "student_ta_r1_768": 0.30140000581741333,
    "student_ta_r5_128": 0.5397999882698059,
    "student_ta_r5_1280": 0.551800012588501,
    "student_ta_r5_256": 0.5529999732971191,
    "student_ta_r5_512": 0.5523999929428101,
    "student_ta_r5_768": 0.5532000064849854,
    "teacher_at_r10_128": 0.7107999920845032,
    "teacher_at_r10_1280": 0.7335999608039856,
    "teacher_at_r10_256": 0.7277999520301819,
    "teacher_at_r10_512": 0.7299999594688416,
    "teacher_at_r10_768": 0.7333999872207642,
    "teacher_at_r1_128": 0.35819998383522034,
    "teacher_at_r1_1280": 0.3946000039577484,
    "teacher_at_r1_256": 0.3929999768733978,
    "teacher_at_r1_512": 0.3953999876976013,
    "teacher_at_r1_768": 0.3951999843120575,
    "teacher_at_r5_128": 0.6187999844551086,
    "teacher_at_r5_1280": 0.642799973487854,
    "teacher_at_r5_256": 0.6367999911308289,
    "teacher_at_r5_512": 0.640999972820282,
    "teacher_at_r5_768": 0.6407999992370605
  },
  "lora": {
    "alpha": 16.0,
    "dropout": 0.0,
    "rank": 0,
    "targets": []
  },
  "merged_lora_source": "triembed/checkpoints/mn20_native_lora_aistmix_audioheavy100k175k175k_continue_from_balanced_20260426T143137Z/latest_model.pt",
  "modalities": [
    "audio",
    "speech"
  ],
  "model_id": "AS-20M",
  "size_millions_rounded": 20,
  "source_checkpoint": "triembed/checkpoints/mn20_native_merged_aistmix_audioheavy100k175k175k_continue_from_balanced_20260426T143137Z/latest_model.pt",
  "source_checkpoint_sha256": "f43003f4d8dbc1eaa0095e1f3cab608ecca3309e77f579e5078c269c899ade52"
}