schedule.json

{"base_lr": [{"loc": 0.0008, "scale": 0.0001}], "base_lr_sync": [true], "base_bs": [16], "base_lr_scale": ["sqrt"], "batch_size": [10], "lr": [null], "checkpoint": [null], "in_channels": [3], "model": ["CpnResNeXt101UNet"], "pretrained": [false], "encoder_kwargs": [{}], "backbone_kwargs": [{"pyr_pool": {"out_channels": 256, "scales": [1, 2, 3, 6], "method": "ppm"}}], "sync_batch_norm": [true], "bg_fg_dists": [[0.75, 0.75]], "order": [25], "nms_thresh": [0.3141592653589793], "score_thresh": [0.8], "certainty_thresh": [null], "samples": [224], "refinement": [true], "refinement_iterations": [4], "refinement_margin": [3.0], "refinement_buckets": [6], "contour_features": ["1"], "refinement_features": ["0"], "contour_head_stride": [1], "order_weights": [true], "refinement_head_stride": [1], "refinement_interpolation": ["bilinear"], "uncertainty_head": [true], "uncertainty_factor": [7.0], "uncertainty_nms": [true], "decoder_activation": ["LeakyReLU"], "decoder_group_norm": [true], "decoder_block": ["ResBlock"], "loss_fourier": ["l1"], "loss_regression": ["l1"], "loss_iou": ["giou"], "loss_classification": ["bce"], "pseudo_labels": [null], "pseudo_labels_num": [null], "aug_plan": ["mild"], "dyn_resize_min": [4], "dyn_resize_max": [192], "dyn_resize_aspect_std": [0.05], "data_norm_method": ["prov"], "classes": [1], "sampler_seed": [null], "sampler_seed_sync": [true], "epochs": [600], "crop_size": [[512, 512]], "image_std": [[0.229, 0.224, 0.225]], "image_mean": [[0.485, 0.456, 0.406]], "amp": [false], "prefetch_factor": [4], "pin_memory": [true], "shuffle": [true], "show_gpu_stats": [true], "show_progress": [true], "save_frequency": [15], "save_min_epoch": [30], "writer_kwargs": [null], "neurips_reps": [5], "optimizer": [{"Adam": {"betas": [0.9, 0.999], "weight_decay": 2e-05}}], "scheduler": [{"WarmupMultiStepLR": {"warmup_factor": 0.001, "warmup_steps": 1000, "warmup_method": "linear", "gamma": 0.666}}], "scheduler_on_step": [true], "scheduler_milestones_as_fractions": [[0.1, 0.2, 0.3, 0.4, 0.6, 0.8]]}