pretrain.json

{
    "name": "pretrain",
    "cuda": true,
    "gpus": [0, 1],
    "precision": 32,
    "backbone_weights": "./vision-0.9.0",
    "pretrain": "/home/luning/dev/projects/FOTS.PyTorch/saved/pretrain_reg_nb/checkpoints/epoch=3-step=259999.ckpt",
    "freeze_det": false,
    "data_loader": {
        "dataset":"synth800k",
        "data_dir": "/home/luning/dev/data/SynthText800k/detection/",
        "scale": 0.25,
        "train": {
            "batch_size": 8,
            "max_transcripts_per_batch": 16,
            "workers": 4,
            "size": [640, 640]
        },
        "val": {
            "batch_size": 8,
            "max_transcripts_per_batch": 16,
            "workers": 4,
            "size": [640, 640]
        }
    },
    "lr_scheduler": {
        "name": null,
        "args":
        {
            "gamma": 0.1,
            "step_size": 10
        }

    },
 
    "optimizer_type": "AdamW",
    "optimizer": {
        "lr": 0.0001
    },
    "loss": "FOTSLoss",
    "metrics": ["fots_metric"],
    "trainer": {
        "epochs": 10,
        "save_dir": "saved/",
        "save_freq": 1,
        "verbosity": 2,
        "monitor": "loss",
        "monitor_mode": "min",
        "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,
        "save_top_k": 5,
        "every_n_train_steps": 100
    },
    "arch": "FOTSModel",
    "model": {
        "mode": "e2e",
        "recognizer": {
            "dropout": 0.2,
            "max_width": 80
        }
    }
}