configs/base.json

{
    "embed_dim": 384,
    "num_queries": 128,
    "num_encoder_layers": 4,
    "num_decoder_layers": 5,
    "num_scale_layers": 3,

    "weight_coord": 5.0,
    "weight_ce": 1.0,
    "weight_vertex_consistency": 0.5,

    "use_cdn": true,
    "cdn_groups": 5,
    "cdn_lambda_pos": 0.4,
    "cdn_lambda_neg": 0.8,
    "cdn_weight_ce": 1.0,
    "cdn_weight_coord": 5.0,

    "learning_rate": 1e-4,
    "weight_decay": 1e-4,
    "scheduler": "onecycle",
    "clip_grad_norm": 0.1,

    "num_points_train": 4096,
    "num_points_val": 7168,
    "batch_size": 14,
    "num_epochs": 200,
    "pt_type": "colmap",

    "aug_rotate": true,
    "aug_flip": true,
    "precision": "fp16"
}