Hugging Face's logo

AbstractPhil
/
captionbert-8192-v2

Feature Extraction
Transformers
TensorBoard
Safetensors
English
captionbert_v2
sentence-similarity
consensus-distillation
geometric-deep-learning
amoe
custom_code
Model card Files
xet
 Metrics Community
captionbert-8192-v2
File size: 1,658 Bytes
9154293
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
81
82
83
84
85
{
  "run_name": "captionbert-v2-moe",
  "trunk_repo": "AbstractPhil/captionbert-8192-v2",
  "trunk_ckpt": "checkpoints/best_model.pt",
  "tokenizer": "google-bert/bert-base-uncased",
  "d_model": 512,
  "n_heads": 8,
  "n_layers": 12,
  "d_ff": 2048,
  "max_len": 8192,
  "output_dim": 768,
  "pooling": "mean",
  "experts": [
    [
      "equiv",
      [
        [
          "sentence-transformers/all-nli",
          "triplet",
          "anchor",
          "positive",
          "negative",
          200000
        ]
      ]
    ],
    [
      "simplify",
      [
        [
          "sentence-transformers/simple-wiki",
          "pair",
          "text",
          "simplified",
          null,
          0
        ],
        [
          "sentence-transformers/altlex",
          "pair",
          "text",
          "simplified",
          null,
          0
        ],
        [
          "sentence-transformers/sentence-compression",
          "pair",
          "text",
          "simplified",
          null,
          0
        ]
      ]
    ]
  ],
  "dedup_jaccard": 0.95,
  "n_slots": 16,
  "K": 64,
  "D": 4,
  "tau": 0.1,
  "hidden": 178,
  "gate_init": -3.0,
  "anchor_steps": 1500,
  "anchor_lr": 0.001,
  "batch_size": 256,
  "temperature": 0.05,
  "max_tokens": 64,
  "align_steps": 800,
  "align_lr": 0.001,
  "align_emb": 64,
  "align_tau": 0.1,
  "check_every": 200,
  "usage_ppl_floor": 1.5,
  "usage_min": 0.02,
  "max_strikes": 3,
  "seed": 0,
  "log_every": 100,
  "eval_every": 500,
  "out_dir": "/content/amoe_moe",
  "hf_repo": "AbstractPhil/captionbert-8192-v2",
  "hf_path": "amoe/moe",
  "hf_private": false,
  "hf_push": true
}