config.json

{
    "dictionary": "keys.json",
    "name": "FOTS",
    "cuda": true,
    "gpus": [1],
    "data_loader": {
        "dataset":"synth800k",
        "data_dir": "/Users/macbook/Documents/Cinnamon/E2E/dataset/data_v1",
        "batch_size": 5,
        "shuffle": true,
        "workers": 0
    },
    "validation": {
        "validation_split": 0.2,
        "shuffle": true, 
        "val_interval": 2
    },
    "lr_scheduler_type": "",
    "lr_scheduler_freq": 10000,
    "lr_scheduler": {
            "gamma": 0.94
    },
 
    "optimizer_type": "Adam",
    "optimizer": {
        "lr": 0.001
    },
    "loss": "FOTSLoss",
    "metrics": ["fots_metric"],
    "trainer": {
        "epochs": 5,
        "save_dir": "saved_models/",
        "save_freq": 1,
        "verbosity": 2,
        "monitor": "hmean",
        "monitor_mode": "max"
    },
    "arch": "FOTSModel",
    "model": {
        "mode": "united"
    }
}