UI-LIC/example-arguments.json at main · BaylorMultimediaLab/UI-LIC · 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
{
    "global_arguments": {
        "cuda": true,
        "gpu_id": 0,
        "epochs": 5,
        "learning_rate": 0.00067,
        "batch_size": 16,
        "test_batch_size": 1,
        "patch_size": [256, 256],
        "train_dataset": "~/data/train",
        "test_dataset": "~/data/test"
    },
    "tasks": {
        "DCVC": {
            "env_path": "training-testing-env",
            "task_name": "DCVC-RT",
            "directory": "DCVC-Fork",
            "arguments": {
                "experiment": "interactive_run_01",
                "save": true
            }
        },
        "ELIC": {
            "env_path": "training-testing-env",
            "task_name": "ELIC",
            "directory": "ELIC",
            "arguments": {
                "experiment": "interactive_run_01",
                "save": true
            }
        },
        "StableCodec": {
            "env_path": "~/training-testing-env",
            "task_name": "StableCodec",
            "directory": "StableCodec",
            "arguments": {
                "experiment": "interactive_run_01",
                "save": true,
                "sd_path": "~/sd-turbo",
                "elic_path": "~/data/elic_official.pth",
                "lambda_rate": "0.01"
            }
        },
        "LIC-HPCM": {
            "env_path": "~/HPCM-training-testing-env",
            "task_name": "LIC-HPCM",
            "directory": "LIC-HPCM",
            "arguments": {
                "model_name": "HPCM_Base",
                "train_dataset": "~/data/train",
                "test_dataset": "~/data/test",
                "experiment": "interactive_run_01",
                "save": true
            }
        }
    }
}