config.json

{
    "name": "CDS-MVSNet",
    "n_gpu": 1,
    "arch": {
        "type": "CDSMVSNet",
        "args": {
            "refine": true,
            "ndepths": [
                48,
                32,
                8
            ],
            "depth_interals_ratio": [
                4.0,
                1.5,
                0.75
            ],
            "share_cr": false,
            "cr_base_chs": [
                8,
                8,
                8
            ],
            "grad_method": "detach"
        }
    },
    "data_loader": [
        {
            "type": "DTULoader",
            "args": {
                "data_path": "/mnt/sdb/khang/dtu_dataset/train",
                "train_data_list": "lists/dtu/train.txt",
                "val_data_list": "lists/dtu/val.txt",
                "mode": "train",
                "num_srcs": 3,
                "num_depths": 192,
                "interval_scale": 1.06,
                "shuffle": true,
                "seq_size": 7,
                "batch_size": 6
            }
        },
        {
            "type": "BlendedLoader",
            "args": {
                "data_path": "/mnt/sdb/khang/blendedmvs",
                "train_data_list": "lists/blended/training_list.txt",
                "val_data_list": "lists/blended/validation_list.txt",
                "mode": "train",
                "num_srcs": 3,
                "num_depths": 192,
                "interval_scale": 1.0,
                "shuffle": true,
                "seq_size": 7,
                "batch_size": 6
            }
        }
    ],
    "optimizer": {
        "type": "SGD",
        "args": {
            "lr": 0.01,
            "weight_decay": 0.01
        }
    },
    "loss": "final_loss",
    "metrics": [
        "mae",
        "rmse",
        "mre"
    ],
    "lr_scheduler": {
        "type": "StepLR",
        "args": {
            "step_size": 3,
            "gamma": 0.5
        }
    },
    "trainer": {
        "epochs": 30,
        "seq": true,
        "save_dir": "saved/",
        "save_period": 1,
        "verbosity": 2,
        "logging_every": 100,
        "early_stop": 20,
        "tensorboard": false,
        "depth_scale": 1.0,
        "lrepochs": "5,10,15:2",
        "dlossw": [
            0.5,
            1.0,
            2.0
        ],
        "eval_freq": 1
    }
}