File size: 1,406 Bytes
14ea361
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "model": {
    "num_labels": 1641,
    "max_len": 256,
    "d_model": 192,
    "n_layers": 3,
    "n_heads": 3,
    "dropout": 0.1,
    "conv_kernel": 5,
    "downsample": 1,
    "ngram_buckets": 4096,
    "script_buckets": 512,
    "rope_theta": 10000.0,
    "vocab_size": 259,
    "modalities": [
      "text"
    ],
    "model_type": "millid-byte-encoder"
  },
  "training": {
    "dataset": "baobabtech/millid-dataset",
    "preset": "tiny",
    "max_len": 256,
    "steps": 120000,
    "batch_size": 512,
    "lr": 0.001,
    "warmup": 2000,
    "weight_decay": 0.01,
    "dropout": 0.1,
    "label_smoothing": 0.1,
    "alpha": 0.3,
    "short_p": 0.25,
    "eval_every": 2000,
    "eval_per_label": 20,
    "patience": 12000,
    "commercial_only": false,
    "extra": [],
    "compare_glotlid": true,
    "compare_fasttext": true,
    "final_eval": "all",
    "output_dir": "millid-out",
    "push_to": "baobabtech/millid-tiny-v1.5",
    "public": false,
    "seed": 7,
    "log_every": 100,
    "threads": 0,
    "workers": -1,
    "compile": true,
    "gpus": 0,
    "wandb_project": "millid",
    "wandb_entity": "mscsr000324-must",
    "wandb_run_name": "millid-tiny-v1.5",
    "wandb_tags": [
      "tiny",
      "v1.5"
    ],
    "no_wandb": false
  },
  "best_step": 106000,
  "parameters": 2927145,
  "gpus": 1,
  "train_passages": 1943957,
  "trained_labels": 1641,
  "seconds": 1999
}