collect_input_args = [
    'image_lr',
    'crops_image_hr',
    'depth_gt',
    'crop_depths',
    'bboxs',
    'image_hr',
]
convert_syncbn = True
debug = False
env_cfg = dict(
    cudnn_benchmark=True,
    dist_cfg=dict(backend='nccl'),
    mp_cfg=dict(mp_start_method='forkserver'))
find_unused_parameters = True
general_dataloader = dict(
    batch_size=1,
    dataset=dict(
        dataset_name='', gt_dir=None, rgb_image_dir='', type='ImageDataset'),
    num_workers=2)
launcher = 'pytorch'
log_name = 'patchfusion'
max_depth = 80
min_depth = 0.001
model = dict(
    coarse_branch=dict(
        attractor_alpha=1000,
        attractor_gamma=2,
        attractor_kind='mean',
        attractor_type='inv',
        aug=True,
        bin_centers_type='softplus',
        bin_embedding_dim=128,
        clip_grad=0.1,
        dataset='nyu',
        distributed=True,
        do_resize=False,
        force_keep_ar=True,
        freeze_midas_bn=True,
        gpu='NULL',
        img_size=[
            384,
            512,
        ],
        inverse_midas=False,
        log_images_every=0.1,
        max_depth=80,
        max_temp=50.0,
        max_translation=100,
        memory_efficient=True,
        midas_model_type='DPT_BEiT_L_384',
        min_depth=0.001,
        min_temp=0.0212,
        model='zoedepth',
        n_attractors=[
            16,
            8,
            4,
            1,
        ],
        n_bins=64,
        name='ZoeDepth',
        notes='',
        output_distribution='logbinomial',
        prefetch=False,
        pretrained_resource='local::./work_dir/ZoeDepthv1.pt',
        print_losses=False,
        project='ZoeDepth',
        random_crop=False,
        random_translate=False,
        root='.',
        save_dir='',
        shared_dict='NULL',
        tags='',
        train_midas=True,
        translate_prob=0.2,
        type='ZoeDepth',
        uid='NULL',
        use_amp=False,
        use_pretrained_midas=True,
        use_shared_dict=False,
        validate_every=0.25,
        version_name='v1',
        workers=16),
    fine_branch=dict(
        attractor_alpha=1000,
        attractor_gamma=2,
        attractor_kind='mean',
        attractor_type='inv',
        aug=True,
        bin_centers_type='softplus',
        bin_embedding_dim=128,
        clip_grad=0.1,
        dataset='nyu',
        distributed=True,
        do_resize=False,
        force_keep_ar=True,
        freeze_midas_bn=True,
        gpu='NULL',
        img_size=[
            384,
            512,
        ],
        inverse_midas=False,
        log_images_every=0.1,
        max_depth=80,
        max_temp=50.0,
        max_translation=100,
        memory_efficient=True,
        midas_model_type='DPT_BEiT_L_384',
        min_depth=0.001,
        min_temp=0.0212,
        model='zoedepth',
        n_attractors=[
            16,
            8,
            4,
            1,
        ],
        n_bins=64,
        name='ZoeDepth',
        notes='',
        output_distribution='logbinomial',
        prefetch=False,
        pretrained_resource='local::./work_dir/ZoeDepthv1.pt',
        print_losses=False,
        project='ZoeDepth',
        random_crop=False,
        random_translate=False,
        root='.',
        save_dir='',
        shared_dict='NULL',
        tags='',
        train_midas=True,
        translate_prob=0.2,
        type='ZoeDepth',
        uid='NULL',
        use_amp=False,
        use_pretrained_midas=True,
        use_shared_dict=False,
        validate_every=0.25,
        version_name='v1',
        workers=16),
    guided_fusion=dict(g2l=True, n_channels=5, type='GuidedFusionPatchFusion'),
    max_depth=80,
    min_depth=0.001,
    pretrain_model=[
        './work_dir/coarse_pretrain/checkpoint_24.pth',
        './work_dir/fine_pretrain/checkpoint_24.pth',
    ],
    sigloss=dict(type='SILogLoss'),
    type='PatchFusion')
optim_wrapper = dict(
    clip_grad=dict(max_norm=0.1, norm_type=2, type='norm'),
    optimizer=dict(lr=0.0001, type='AdamW', weight_decay=0.001),
    paramwise_cfg=dict(bypass_duplicate=True, custom_keys=dict()))
param_scheduler = dict(
    base_momentum=0.85,
    cycle_momentum=True,
    div_factor=10,
    final_div_factor=10000,
    max_momentum=0.95,
    pct_start=0.25,
    three_phase=False)
project = 'patchfusion'
resume = False
tags = [
    'patchfusion',
]
test_in_dataloader = dict(
    batch_size=1,
    dataset=dict(
        data_root='./data/u4k',
        max_depth=80,
        min_depth=0.001,
        mode='infer',
        split='./data/u4k/splits/test.txt',
        transform_cfg=dict(network_process_size=[
            384,
            512,
        ]),
        type='UnrealStereo4kDataset'),
    num_workers=2)
test_out_dataloader = dict(
    batch_size=1,
    dataset=dict(
        data_root='./data/u4k',
        max_depth=80,
        min_depth=0.001,
        mode='infer',
        split='./data/u4k/splits/test_out.txt',
        transform_cfg=dict(network_process_size=[
            384,
            512,
        ]),
        type='UnrealStereo4kDataset'),
    num_workers=2)
train_cfg = dict(
    eval_start=0,
    log_interval=100,
    max_epochs=16,
    save_checkpoint_interval=16,
    train_log_img_interval=500,
    val_interval=2,
    val_log_img_interval=10,
    val_type='epoch_base')
train_dataloader = dict(
    batch_size=4,
    dataset=dict(
        data_root='./data/u4k',
        max_depth=80,
        min_depth=0.001,
        mode='train',
        split='./data/u4k/splits/train.txt',
        transform_cfg=dict(
            degree=1.0,
            network_process_size=[
                384,
                512,
            ],
            random_crop=True,
            random_crop_size=(
                540,
                960,
            )),
        type='UnrealStereo4kDataset'),
    num_workers=4)
val_dataloader = dict(
    batch_size=1,
    dataset=dict(
        data_root='./data/u4k',
        max_depth=80,
        min_depth=0.001,
        mode='infer',
        split='./data/u4k/splits/val.txt',
        transform_cfg=dict(
            network_process_size=[
                384,
                512,
            ], random_crop_size=(
                540,
                960,
            )),
        type='UnrealStereo4kDataset'),
    num_workers=2)
work_dir = './work_dir/patchfusion'
zoe_depth_config = dict(
    attractor_alpha=1000,
    attractor_gamma=2,
    attractor_kind='mean',
    attractor_type='inv',
    aug=True,
    bin_centers_type='softplus',
    bin_embedding_dim=128,
    clip_grad=0.1,
    dataset='nyu',
    distributed=True,
    do_resize=False,
    force_keep_ar=True,
    freeze_midas_bn=True,
    gpu='NULL',
    img_size=[
        384,
        512,
    ],
    inverse_midas=False,
    log_images_every=0.1,
    max_depth=80,
    max_temp=50.0,
    max_translation=100,
    memory_efficient=True,
    midas_model_type='DPT_BEiT_L_384',
    min_depth=0.001,
    min_temp=0.0212,
    model='zoedepth',
    n_attractors=[
        16,
        8,
        4,
        1,
    ],
    n_bins=64,
    name='ZoeDepth',
    notes='',
    output_distribution='logbinomial',
    prefetch=False,
    pretrained_resource='local::./work_dir/ZoeDepthv1.pt',
    print_losses=False,
    project='ZoeDepth',
    random_crop=False,
    random_translate=False,
    root='.',
    save_dir='',
    shared_dict='NULL',
    tags='',
    train_midas=True,
    translate_prob=0.2,
    type='ZoeDepth',
    uid='NULL',
    use_amp=False,
    use_pretrained_midas=True,
    use_shared_dict=False,
    validate_every=0.25,
    version_name='v1',
    workers=16)