train_e2fgvi_hq.json

{
    "seed": 2021,
    "save_dir": "release_model/",
    "train_data_loader": {
        "name": "youtube-vos",
        "data_root": "datasets",
        "w": 432,
        "h": 240,
        "num_local_frames": 5,
        "num_ref_frames": 3
    },
    "losses": {
        "hole_weight": 1,
        "valid_weight": 1,
        "flow_weight": 1,
        "adversarial_weight": 0.01,
        "GAN_LOSS": "hinge"
    },
    "model": {
        "net": "e2fgvi_hq",
        "no_dis": 0
    },
    "trainer": {
        "type": "Adam",
        "beta1": 0,
        "beta2": 0.99,
        "lr": 1e-4,
        "batch_size": 8,
        "num_workers": 2,
        "log_freq": 100,
        "save_freq": 5e3,
        "iterations": 50e4,
        "scheduler": {
            "type": "MultiStepLR",
            "milestones": [
                40e4
            ],
            "gamma": 0.1
        }
    }
}