collaborative-experts/configs/data_loader_lsmdc.json at master · Deferf/collaborative-experts · 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
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
{
    "inherit_from": "configs/base_config.json",
    "eval_mode": "test_run",
    "experts": {
        "face_dim": 128,
        "text_dim": 300,
        "text_feat": "w2v",
        "modalities": [
            "imagenet.resnext101_32x48d.0",
            "imagenet.senet154.0",
            "scene.densenet161.0",
            "r2p1d.r2p1d-ig65m.0",
            "i3d.i3d.0",
            "ocr",
            "audio",
            "speech",
            "face"
        ]
    },
    "arch": {
        "args": {
            "test_caption_mode": "indep",
            "use_ce": "pairwise",
            "use_mish": 1,
            "use_bn_reason": 1,
            "num_g_layers": 3,
            "num_h_layers": 0,
            "include_self": 1,
            "l2renorm": false,
            "vlad_clusters": {
                "ocr": 30,
                "text": 43,
                "audio": 3,
                "speech": 5,
                "detection-sem": 50
            },
            "ghost_clusters": {
                "text": 0,
                "ocr": 0,
                "audio": 2,
                "speech": 2
            },
            "mimic_ce_dims": 0
        }
    },
    "data_loader": {
        "args":{
            "dataset_name": "LSMDC",
            "data_dir": "data/LSMDC",
            "split_name": "full-val",
            "batch_size": 256,
            "fuse_captions": false,
            "num_test_captions": 1,
            "max_tokens": {
                "ocr": 6,
                "text": 36,
                "speech": 35,
                "audio": 11,
                "detection": 12,
                "detection-sem": 100,
                "openpose": 50
            }
        }
    },
    "visualizer": {
        "args":{
            "src_video_dir": "data/LSMDC/videos",
            "vis_vid_freq": 1000,
            "num_samples": 1
        }
    },
    "optimizer": {
        "type": "Ranger",
        "args":{
            "lr": 0.01,
            "weight_decay": 1e-3
        }
    },
    "loss": {
        "type": "MaxMarginRankingLoss",
        "args":{
            "margin": 0.12132983763957966,
            "fix_norm": true
        }
    },
    "trainer": {
        "epochs": 15
    },
    "eval_settings": {
        "data_loader": {
            "args":{
                "split_name": "full-test",
                "num_test_captions": 1
            }
        },
        "tester": {
            "save_dir": "data/saved/",
            "verbosity": 2
        },
        "disable_gpu": true
    }
}