roco-image-captioning/config_generator.py at main · fcoelhomrc/roco-image-captioning · 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
import yaml

config = {
    "dataset": {
        "train_json": "/home/felipe/Projects/roco-image-captioning/roco-dataset/json/train_dataset.json",
        "validation_json": "/home/felipe/Projects/roco-image-captioning/roco-dataset/json/validation_dataset.json",
        "test_json": "/home/felipe/Projects/roco-image-captioning/roco-dataset/json/test_dataset.json",
    },
    "input_size": [224, 224],
    "train": {
        "max_seq_length": 70,
        "max_epochs": 10,
        "batch_size": 4,
        'limit_train_batches': 0.10,
        'early_stopping_patience': 5,
        'log_every_n_steps': 4,
        "gradient_clip_val": 10.0,
    },
    "optimizer": {
        'method': 'adam',
        'weight_decay': 5e-4,
        'lr': 1e-5,
        "lr_scheduler_rate": 0.1,
        "lr_scheduler_patience": 1,
    },
    "registry": {
        'project': "roco-debug",
        'name': "roco-clip",
        'wandb_root_dir': "outputs/wandb_outputs",
        'checkpoints_dir': "outputs/model_checkpoints",
    }
}

with open("config.yaml", "w") as outfile:
    yaml.dump(config, outfile, default_flow_style=False)

from pprint import pprint
pprint(config)