pretrain.json

{
    "name": "pretrain",
    "cuda": true,
    "gpus": [0, 1],
    "precision": 32,
    "backbone_weights": "/home/luning/dev/projects/FOTS.PyTorch/vision-0.9.0",
    "pretrain": "",
    "data_loader": {
        "dataset":"synth800k",
        "data_dir": "/home/luning/dev/data/SynthText800k/detection/",
        "batch_size": 8,
        "shuffle": true,
        "workers": 8,
        "max_transcripts_pre_batch": 32,
        "size": 640,
        "scale": 0.25
    },
    "validation": {

        "scales": [800, 1000, 1333]
    },
    "lr_scheduler": {
        "name": "StepLR",
        "args":
        {
            "gamma": 0.1,
            "step_size": 5
        }

    },
 
    "optimizer_type": "AdamW",
    "optimizer": {
        "lr": 0.0001
    },
    "loss": "FOTSLoss",
    "metrics": ["fots_metric"],
    "trainer": {
        "epochs": 10,
        "save_dir": "saved/",
        "save_freq": 1,
        "verbosity": 2,
        "monitor": "hmean",
        "monitor_mode": "max",
        "log_every_n_steps": 1,
        "log_gpu_memory": "all",
        "overfit_batches": 0.0,
        "terminate_on_nan": false,
        "fast_dev_run": false,
        "check_val_every_n_epoch": 1
    },
    "arch": "FOTSModel",
    "model": {
        "mode": "united"
    }
}