config_ada.json

{
    "name": "ResNetAE_RGB100",
    "n_gpu": 4,

    "arch": {
        "type": "ResNetAE_RGB",
        "args": {"n_levels": 4,
            "n_ResidualBlock": 4,
            "z_dim": 128,
            "UseMultiResSkips": true}
    },
    "dataloader": {
        "type": "TemporalDataloader",
        "args":{
            "dataset_path": "/../../../ssd_scratch/cvit/richa/dataset/",
            "seq_ID" : "03384",
            "K_frames": 4,
            "batch_size": 3,
            "shuffle": true,
            "validation_split": 0.0,
            "num_workers": 2
        }
    },
    "optimizer": {
        "type": "Adam",
        "args":{
            "lr": 0.001,
            "weight_decay": 0.01,
            "amsgrad": true
        }
    },
    "loss": "ssim_L1_loss",
    "metrics": [
    ],
    "lr_scheduler": {
        "type": "ExponentialLR",
        "args": {
            "gamma": 0.99
        }
    },
    "trainer": {
        "epochs": 100,

        "save_dir": "/../../../ssd_scratch/cvit/richa/saved/",
        "save_period": 10,
        "verbosity": 2,
        
        "monitor": "off",
        "early_stop": 10,

        "tensorboard": true
    }
}