rotlex/conf.json at main · ustcllm/rotlex · GitHub

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
{
    "dataset_name": "Webvid",
    "used_label_num": 10,
    "train_set_len": -1,
    "self_train_set_len": -1,
    "net_type": "deepnet_sep",
    "mlp_wth": [
        512
    ],
    "code_emb_dim": 512,
    "device": "cuda:0",
    "bn": true,
    "drop": 1000000000.0,
    "activate": "swish",
    "k": 128,
    "num_layers": 2,
    "R": 1,
    "ivf_device": "cuda:0",
    "epoch": 400,
    "lr": 0.001,
    "bs": 5000,
    "val_bs": 100,
    "es_patient": 40,
    "layer_weight": [
        1,
        1
    ],
    "sample_thres": 20000,
    "sample_num": 327,
    "val_interval": 5,
    "upd_interval": 20,
    "upd_patient": 1000000,
    "load_best_upd": false,
    "balance_factor": 1.5,
    "norm_query": false,
    "spherical": false,
    "init": "kmeans",
    "max_iter": 1,
    "reinit_after_upd": false,
    "train_mode": "all",
    "upd_log_softmax": true,
    "upd_norm": false,
    "upd_method": "rotlex",
    "upd_assign_mcmf": true,
    "upd_on_query": true,
    "reconstruct": false,
    "load_tree": "",
    "load_all": "",
    "extra": "",
    "eval_topk": 100,
    "eval_num": 10000,
    "eval_beam": 100
}