├── Ablation ├── Block_Only │ ├── .gitignore │ ├── mamba-1p1p1 │ │ ├── AUTHORS │ │ ├── LICENSE │ │ ├── README.md │ │ ├── assets │ │ │ └── selection.png │ │ ├── benchmarks │ │ │ └── benchmark_generation_mamba_simple.py │ │ ├── csrc │ │ │ └── selective_scan │ │ │ │ ├── reverse_scan.cuh │ │ │ │ ├── selective_scan.cpp │ │ │ │ ├── selective_scan.h │ │ │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ │ │ ├── selective_scan_bwd_bf16_real.cu │ │ │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ │ │ ├── selective_scan_bwd_fp16_real.cu │ │ │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ │ │ ├── selective_scan_bwd_fp32_real.cu │ │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ │ ├── selective_scan_common.h │ │ │ │ ├── selective_scan_fwd_bf16.cu │ │ │ │ ├── selective_scan_fwd_fp16.cu │ │ │ │ ├── selective_scan_fwd_fp32.cu │ │ │ │ ├── selective_scan_fwd_kernel.cuh │ │ │ │ ├── static_switch.h │ │ │ │ └── uninitialized_copy.cuh │ │ ├── evals │ │ │ └── lm_harness_eval.py │ │ ├── mamba_ssm │ │ │ ├── __init__.py │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ ├── config_mamba.py │ │ │ │ └── mixer_seq_simple.py │ │ │ ├── modules │ │ │ │ ├── __init__.py │ │ │ │ └── mamba_simple.py │ │ │ ├── ops │ │ │ │ ├── __init__.py │ │ │ │ ├── selective_scan_interface.py │ │ │ │ └── triton │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── layernorm.py │ │ │ │ │ └── selective_state_update.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── generation.py │ │ │ │ └── hf.py │ │ ├── setup.py │ │ └── tests │ │ │ └── ops │ │ │ ├── test_selective_scan.py │ │ │ └── triton │ │ │ └── test_selective_state_update.py │ └── vim │ │ ├── LICENSE │ │ ├── augment.py │ │ ├── calc_flops.py │ │ ├── calc_throughput.py │ │ ├── datasets.py │ │ ├── engine.py │ │ ├── hubconf.py │ │ ├── losses.py │ │ ├── main.py │ │ ├── models │ │ ├── embedding.py │ │ ├── mamba_block.py │ │ ├── predictor.py │ │ └── vim_block_only.py │ │ ├── rope.py │ │ ├── samplers.py │ │ ├── scripts │ │ ├── vim-b.sh │ │ ├── vim-s.sh │ │ └── vim-t.sh │ │ ├── utils.py │ │ └── vim_requirements.txt ├── Predictor_Input │ ├── .gitignore │ ├── Predictor-B │ │ ├── .gitignore │ │ ├── =1.1.0 │ │ ├── LICENSE.txt │ │ ├── README.md │ │ ├── assets │ │ │ ├── vim_pipeline_v1.9.png │ │ │ └── vim_teaser_v1.7.png │ │ ├── det │ │ │ ├── .circleci │ │ │ │ ├── config.yml │ │ │ │ └── import-tests.sh │ │ │ ├── .clang-format │ │ │ ├── .flake8 │ │ │ ├── .gitignore │ │ │ ├── GETTING_STARTED.md │ │ │ ├── INSTALL.md │ │ │ ├── LICENSE │ │ │ ├── MODEL_ZOO.md │ │ │ ├── README.md │ │ │ ├── configs │ │ │ │ ├── Base-RCNN-C4.yaml │ │ │ │ ├── Base-RCNN-DilatedC5.yaml │ │ │ │ ├── Base-RCNN-FPN.yaml │ │ │ │ ├── Base-RetinaNet.yaml │ │ │ │ ├── COCO-Detection │ │ │ │ │ ├── fast_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── faster_rcnn_R_101_C4_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_101_DC5_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_101_FPN_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_C4_1x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_C4_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_DC5_1x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_DC5_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_FPN_3x.yaml │ │ │ │ │ ├── faster_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ │ ├── fcos_R_50_FPN_1x.py │ │ │ │ │ ├── retinanet_R_101_FPN_3x.yaml │ │ │ │ │ ├── retinanet_R_50_FPN_1x.py │ │ │ │ │ ├── retinanet_R_50_FPN_1x.yaml │ │ │ │ │ ├── retinanet_R_50_FPN_3x.yaml │ │ │ │ │ ├── rpn_R_50_C4_1x.yaml │ │ │ │ │ └── rpn_R_50_FPN_1x.yaml │ │ │ │ ├── COCO-InstanceSegmentation │ │ │ │ │ ├── mask_rcnn_R_101_C4_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_101_DC5_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_101_FPN_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_1x.py │ │ │ │ │ ├── mask_rcnn_R_50_C4_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_DC5_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_DC5_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_giou.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x.yaml │ │ │ │ │ ├── mask_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_fpn_1x.py │ │ │ │ │ └── mask_rcnn_regnety_4gf_dds_fpn_1x.py │ │ │ │ ├── COCO-Keypoints │ │ │ │ │ ├── Base-Keypoint-RCNN-FPN.yaml │ │ │ │ │ ├── keypoint_rcnn_R_101_FPN_3x.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.py │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_3x.yaml │ │ │ │ │ └── keypoint_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ ├── COCO-PanopticSegmentation │ │ │ │ │ ├── Base-Panoptic-FPN.yaml │ │ │ │ │ ├── panoptic_fpn_R_101_3x.yaml │ │ │ │ │ ├── panoptic_fpn_R_50_1x.py │ │ │ │ │ ├── panoptic_fpn_R_50_1x.yaml │ │ │ │ │ └── panoptic_fpn_R_50_3x.yaml │ │ │ │ ├── Cityscapes │ │ │ │ │ └── mask_rcnn_R_50_FPN.yaml │ │ │ │ ├── Detectron1-Comparisons │ │ │ │ │ ├── README.md │ │ │ │ │ ├── faster_rcnn_R_50_FPN_noaug_1x.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ └── mask_rcnn_R_50_FPN_noaug_1x.yaml │ │ │ │ ├── LVISv0.5-InstanceSegmentation │ │ │ │ │ ├── mask_rcnn_R_101_FPN_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ └── mask_rcnn_X_101_32x8d_FPN_1x.yaml │ │ │ │ ├── LVISv1-InstanceSegmentation │ │ │ │ │ ├── mask_rcnn_R_101_FPN_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ └── mask_rcnn_X_101_32x8d_FPN_1x.yaml │ │ │ │ ├── Misc │ │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_3x.yaml │ │ │ │ │ ├── cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_cls_agnostic.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_dconv_c3-c5.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_dconv_c3-c5.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_gn.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_syncbn.yaml │ │ │ │ │ ├── mmdet_mask_rcnn_R_50_FPN_1x.py │ │ │ │ │ ├── panoptic_fpn_R_101_dconv_cascade_gn_3x.yaml │ │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_3x_gn.yaml │ │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_9x_gn.yaml │ │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_9x_syncbn.yaml │ │ │ │ │ ├── semantic_R_50_FPN_1x.yaml │ │ │ │ │ └── torchvision_imagenet_R_50.py │ │ │ │ ├── PascalVOC-Detection │ │ │ │ │ ├── faster_rcnn_R_50_C4.yaml │ │ │ │ │ └── faster_rcnn_R_50_FPN.yaml │ │ │ │ ├── common │ │ │ │ │ ├── README.md │ │ │ │ │ ├── coco_schedule.py │ │ │ │ │ ├── data │ │ │ │ │ │ ├── coco.py │ │ │ │ │ │ ├── coco_keypoint.py │ │ │ │ │ │ ├── coco_panoptic_separated.py │ │ │ │ │ │ └── constants.py │ │ │ │ │ ├── models │ │ │ │ │ │ ├── cascade_rcnn.py │ │ │ │ │ │ ├── fcos.py │ │ │ │ │ │ ├── keypoint_rcnn_fpn.py │ │ │ │ │ │ ├── mask_rcnn_c4.py │ │ │ │ │ │ ├── mask_rcnn_fpn.py │ │ │ │ │ │ ├── mask_rcnn_vimdet.py │ │ │ │ │ │ ├── mask_rcnn_vitdet.py │ │ │ │ │ │ ├── panoptic_fpn.py │ │ │ │ │ │ └── retinanet.py │ │ │ │ │ ├── optim.py │ │ │ │ │ └── train.py │ │ │ │ ├── new_baselines │ │ │ │ │ ├── mask_rcnn_R_101_FPN_100ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_101_FPN_200ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_101_FPN_400ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_100ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_200ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_400ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_50ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_100ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_200ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_400ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnety_4gf_dds_FPN_100ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnety_4gf_dds_FPN_200ep_LSJ.py │ │ │ │ │ └── mask_rcnn_regnety_4gf_dds_FPN_400ep_LSJ.py │ │ │ │ └── quick_schedules │ │ │ │ │ ├── README.md │ │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── fast_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── fast_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_normalized_training_acc_test.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_GCV_instant_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_inference_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_instant_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_training_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_DC5_inference_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_pred_boxes_training_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ │ ├── panoptic_fpn_R_50_inference_acc_test.yaml │ │ │ │ │ ├── panoptic_fpn_R_50_instant_test.yaml │ │ │ │ │ ├── panoptic_fpn_R_50_training_acc_test.yaml │ │ │ │ │ ├── retinanet_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── retinanet_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── rpn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── rpn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── semantic_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── semantic_R_50_FPN_instant_test.yaml │ │ │ │ │ └── semantic_R_50_FPN_training_acc_test.yaml │ │ │ ├── datasets │ │ │ │ ├── README.md │ │ │ │ ├── prepare_ade20k_sem_seg.py │ │ │ │ ├── prepare_cocofied_lvis.py │ │ │ │ ├── prepare_for_tests.sh │ │ │ │ └── prepare_panoptic_fpn.py │ │ │ ├── demo │ │ │ │ ├── README.md │ │ │ │ ├── demo.py │ │ │ │ └── predictor.py │ │ │ ├── det-requirements.txt │ │ │ ├── detectron2 │ │ │ │ ├── __init__.py │ │ │ │ ├── checkpoint │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── c2_model_loading.py │ │ │ │ │ ├── catalog.py │ │ │ │ │ └── detection_checkpoint.py │ │ │ │ ├── config │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── compat.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── defaults.py │ │ │ │ │ ├── instantiate.py │ │ │ │ │ └── lazy.py │ │ │ │ ├── data │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── benchmark.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── catalog.py │ │ │ │ │ ├── common.py │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ ├── datasets │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ ├── builtin_meta.py │ │ │ │ │ │ ├── cityscapes.py │ │ │ │ │ │ ├── cityscapes_panoptic.py │ │ │ │ │ │ ├── coco.py │ │ │ │ │ │ ├── coco_panoptic.py │ │ │ │ │ │ ├── lvis.py │ │ │ │ │ │ ├── lvis_v0_5_categories.py │ │ │ │ │ │ ├── lvis_v1_categories.py │ │ │ │ │ │ ├── lvis_v1_category_image_count.py │ │ │ │ │ │ ├── pascal_voc.py │ │ │ │ │ │ └── register_coco.py │ │ │ │ │ ├── detection_utils.py │ │ │ │ │ ├── samplers │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── distributed_sampler.py │ │ │ │ │ │ └── grouped_batch_sampler.py │ │ │ │ │ └── transforms │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── augmentation.py │ │ │ │ │ │ ├── augmentation_impl.py │ │ │ │ │ │ └── transform.py │ │ │ │ ├── engine │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── defaults.py │ │ │ │ │ ├── hooks.py │ │ │ │ │ ├── launch.py │ │ │ │ │ └── train_loop.py │ │ │ │ ├── evaluation │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── cityscapes_evaluation.py │ │ │ │ │ ├── coco_evaluation.py │ │ │ │ │ ├── evaluator.py │ │ │ │ │ ├── fast_eval_api.py │ │ │ │ │ ├── lvis_evaluation.py │ │ │ │ │ ├── panoptic_evaluation.py │ │ │ │ │ ├── pascal_voc_evaluation.py │ │ │ │ │ ├── rotated_coco_evaluation.py │ │ │ │ │ ├── sem_seg_evaluation.py │ │ │ │ │ └── testing.py │ │ │ │ ├── export │ │ │ │ │ ├── README.md │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── api.py │ │ │ │ │ ├── c10.py │ │ │ │ │ ├── caffe2_export.py │ │ │ │ │ ├── caffe2_inference.py │ │ │ │ │ ├── caffe2_modeling.py │ │ │ │ │ ├── caffe2_patch.py │ │ │ │ │ ├── flatten.py │ │ │ │ │ ├── shared.py │ │ │ │ │ ├── torchscript.py │ │ │ │ │ └── torchscript_patch.py │ │ │ │ ├── layers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── aspp.py │ │ │ │ │ ├── batch_norm.py │ │ │ │ │ ├── blocks.py │ │ │ │ │ ├── csrc │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── ROIAlignRotated │ │ │ │ │ │ │ ├── ROIAlignRotated.h │ │ │ │ │ │ │ ├── ROIAlignRotated_cpu.cpp │ │ │ │ │ │ │ └── ROIAlignRotated_cuda.cu │ │ │ │ │ │ ├── box_iou_rotated │ │ │ │ │ │ │ ├── box_iou_rotated.h │ │ │ │ │ │ │ ├── box_iou_rotated_cpu.cpp │ │ │ │ │ │ │ ├── box_iou_rotated_cuda.cu │ │ │ │ │ │ │ └── box_iou_rotated_utils.h │ │ │ │ │ │ ├── cocoeval │ │ │ │ │ │ │ ├── cocoeval.cpp │ │ │ │ │ │ │ └── cocoeval.h │ │ │ │ │ │ ├── cuda_version.cu │ │ │ │ │ │ ├── deformable │ │ │ │ │ │ │ ├── deform_conv.h │ │ │ │ │ │ │ ├── deform_conv_cuda.cu │ │ │ │ │ │ │ └── deform_conv_cuda_kernel.cu │ │ │ │ │ │ ├── nms_rotated │ │ │ │ │ │ │ ├── nms_rotated.h │ │ │ │ │ │ │ ├── nms_rotated_cpu.cpp │ │ │ │ │ │ │ └── nms_rotated_cuda.cu │ │ │ │ │ │ └── vision.cpp │ │ │ │ │ ├── deform_conv.py │ │ │ │ │ ├── losses.py │ │ │ │ │ ├── mask_ops.py │ │ │ │ │ ├── nms.py │ │ │ │ │ ├── roi_align.py │ │ │ │ │ ├── roi_align_rotated.py │ │ │ │ │ ├── rotated_boxes.py │ │ │ │ │ ├── shape_spec.py │ │ │ │ │ └── wrappers.py │ │ │ │ ├── model_zoo │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── model_zoo.py │ │ │ │ ├── modeling │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── anchor_generator.py │ │ │ │ │ ├── backbone │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── backbone.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── fpn.py │ │ │ │ │ │ ├── mvit.py │ │ │ │ │ │ ├── regnet.py │ │ │ │ │ │ ├── resnet.py │ │ │ │ │ │ ├── swin.py │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ ├── vim.py │ │ │ │ │ │ └── vit.py │ │ │ │ │ ├── box_regression.py │ │ │ │ │ ├── matcher.py │ │ │ │ │ ├── meta_arch │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── dense_detector.py │ │ │ │ │ │ ├── fcos.py │ │ │ │ │ │ ├── panoptic_fpn.py │ │ │ │ │ │ ├── rcnn.py │ │ │ │ │ │ ├── retinanet.py │ │ │ │ │ │ └── semantic_seg.py │ │ │ │ │ ├── mmdet_wrapper.py │ │ │ │ │ ├── poolers.py │ │ │ │ │ ├── postprocessing.py │ │ │ │ │ ├── proposal_generator │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── proposal_utils.py │ │ │ │ │ │ ├── rpn.py │ │ │ │ │ │ └── rrpn.py │ │ │ │ │ ├── roi_heads │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── box_head.py │ │ │ │ │ │ ├── cascade_rcnn.py │ │ │ │ │ │ ├── fast_rcnn.py │ │ │ │ │ │ ├── keypoint_head.py │ │ │ │ │ │ ├── mask_head.py │ │ │ │ │ │ ├── roi_heads.py │ │ │ │ │ │ └── rotated_fast_rcnn.py │ │ │ │ │ ├── sampling.py │ │ │ │ │ └── test_time_augmentation.py │ │ │ │ ├── projects │ │ │ │ │ ├── README.md │ │ │ │ │ └── __init__.py │ │ │ │ ├── solver │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build.py │ │ │ │ │ └── lr_scheduler.py │ │ │ │ ├── structures │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── boxes.py │ │ │ │ │ ├── image_list.py │ │ │ │ │ ├── instances.py │ │ │ │ │ ├── keypoints.py │ │ │ │ │ ├── masks.py │ │ │ │ │ └── rotated_boxes.py │ │ │ │ ├── tracking │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_tracker.py │ │ │ │ │ ├── bbox_iou_tracker.py │ │ │ │ │ ├── hungarian_tracker.py │ │ │ │ │ ├── iou_weighted_hungarian_bbox_iou_tracker.py │ │ │ │ │ ├── utils.py │ │ │ │ │ └── vanilla_hungarian_bbox_iou_tracker.py │ │ │ │ └── utils │ │ │ │ │ ├── README.md │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── analysis.py │ │ │ │ │ ├── collect_env.py │ │ │ │ │ ├── colormap.py │ │ │ │ │ ├── comm.py │ │ │ │ │ ├── develop.py │ │ │ │ │ ├── env.py │ │ │ │ │ ├── events.py │ │ │ │ │ ├── file_io.py │ │ │ │ │ ├── logger.py │ │ │ │ │ ├── memory.py │ │ │ │ │ ├── registry.py │ │ │ │ │ ├── serialize.py │ │ │ │ │ ├── testing.py │ │ │ │ │ ├── tracing.py │ │ │ │ │ ├── video_visualizer.py │ │ │ │ │ └── visualizer.py │ │ │ ├── dev │ │ │ │ ├── README.md │ │ │ │ ├── linter.sh │ │ │ │ ├── packaging │ │ │ │ │ ├── README.md │ │ │ │ │ ├── build_all_wheels.sh │ │ │ │ │ ├── build_wheel.sh │ │ │ │ │ ├── gen_install_table.py │ │ │ │ │ ├── gen_wheel_index.sh │ │ │ │ │ └── pkg_helpers.bash │ │ │ │ ├── parse_results.sh │ │ │ │ ├── run_inference_tests.sh │ │ │ │ └── run_instant_tests.sh │ │ │ ├── docker │ │ │ │ ├── Dockerfile │ │ │ │ ├── README.md │ │ │ │ ├── deploy.Dockerfile │ │ │ │ └── docker-compose.yml │ │ │ ├── docs │ │ │ │ ├── .gitignore │ │ │ │ ├── Makefile │ │ │ │ ├── README.md │ │ │ │ ├── _static │ │ │ │ │ └── css │ │ │ │ │ │ └── custom.css │ │ │ │ ├── conf.py │ │ │ │ ├── index.rst │ │ │ │ ├── modules │ │ │ │ │ ├── checkpoint.rst │ │ │ │ │ ├── config.rst │ │ │ │ │ ├── data.rst │ │ │ │ │ ├── data_transforms.rst │ │ │ │ │ ├── engine.rst │ │ │ │ │ ├── evaluation.rst │ │ │ │ │ ├── export.rst │ │ │ │ │ ├── fvcore.rst │ │ │ │ │ ├── index.rst │ │ │ │ │ ├── layers.rst │ │ │ │ │ ├── model_zoo.rst │ │ │ │ │ ├── modeling.rst │ │ │ │ │ ├── solver.rst │ │ │ │ │ ├── structures.rst │ │ │ │ │ └── utils.rst │ │ │ │ ├── notes │ │ │ │ │ ├── benchmarks.md │ │ │ │ │ ├── changelog.md │ │ │ │ │ ├── compatibility.md │ │ │ │ │ ├── contributing.md │ │ │ │ │ └── index.rst │ │ │ │ ├── requirements.txt │ │ │ │ └── tutorials │ │ │ │ │ ├── README.md │ │ │ │ │ ├── augmentation.md │ │ │ │ │ ├── builtin_datasets.md │ │ │ │ │ ├── configs.md │ │ │ │ │ ├── data_loading.md │ │ │ │ │ ├── datasets.md │ │ │ │ │ ├── deployment.md │ │ │ │ │ ├── evaluation.md │ │ │ │ │ ├── extend.md │ │ │ │ │ ├── getting_started.md │ │ │ │ │ ├── index.rst │ │ │ │ │ ├── install.md │ │ │ │ │ ├── lazyconfigs.md │ │ │ │ │ ├── models.md │ │ │ │ │ ├── training.md │ │ │ │ │ └── write-models.md │ │ │ ├── projects │ │ │ │ ├── DeepLab │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ └── Cityscapes-SemanticSegmentation │ │ │ │ │ │ │ ├── Base-DeepLabV3-OS16-Semantic.yaml │ │ │ │ │ │ │ ├── deeplab_v3_R_103_os16_mg124_poly_90k_bs16.yaml │ │ │ │ │ │ │ └── deeplab_v3_plus_R_103_os16_mg124_poly_90k_bs16.yaml │ │ │ │ │ ├── deeplab │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build_solver.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── loss.py │ │ │ │ │ │ ├── lr_scheduler.py │ │ │ │ │ │ ├── resnet.py │ │ │ │ │ │ └── semantic_seg.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── DensePose │ │ │ │ │ ├── README.md │ │ │ │ │ ├── apply_net.py │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN.yaml │ │ │ │ │ │ ├── HRNet │ │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w32_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w40_s1x.yaml │ │ │ │ │ │ │ └── densepose_rcnn_HRFPN_HRNet_w48_s1x.yaml │ │ │ │ │ │ ├── cse │ │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN-Human.yaml │ │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_soft_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_soft_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_soft_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_4k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_i2m_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_m2m_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_4k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_maskonly_24k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_chimps_finetune_4k.yaml │ │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_soft_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC1M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC1_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC2M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC2_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC1M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC1_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC2M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC2_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x_legacy.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC1M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC1_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC2M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC2_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x_legacy.yaml │ │ │ │ │ │ ├── evolution │ │ │ │ │ │ │ ├── Base-RCNN-FPN-Atop10P_CA.yaml │ │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA.yaml │ │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_coarsesegm.yaml │ │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_finesegm.yaml │ │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_uniform.yaml │ │ │ │ │ │ │ └── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_uv.yaml │ │ │ │ │ │ └── quick_schedules │ │ │ │ │ │ │ ├── cse │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_instant_test.yaml │ │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_soft_animals_finetune_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w32_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_TTA_inference_acc_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ │ ├── densepose │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── converters │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── base.py │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ ├── chart_output_hflip.py │ │ │ │ │ │ │ ├── chart_output_to_chart_result.py │ │ │ │ │ │ │ ├── hflip.py │ │ │ │ │ │ │ ├── segm_to_mask.py │ │ │ │ │ │ │ ├── to_chart_result.py │ │ │ │ │ │ │ └── to_mask.py │ │ │ │ │ │ ├── data │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ │ ├── combined_loader.py │ │ │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ │ │ ├── datasets │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ │ ├── chimpnsee.py │ │ │ │ │ │ │ │ ├── coco.py │ │ │ │ │ │ │ │ ├── dataset_type.py │ │ │ │ │ │ │ │ └── lvis.py │ │ │ │ │ │ │ ├── image_list_dataset.py │ │ │ │ │ │ │ ├── inference_based_loader.py │ │ │ │ │ │ │ ├── meshes │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ │ └── catalog.py │ │ │ │ │ │ │ ├── samplers │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── densepose_base.py │ │ │ │ │ │ │ │ ├── densepose_confidence_based.py │ │ │ │ │ │ │ │ ├── densepose_cse_base.py │ │ │ │ │ │ │ │ ├── densepose_cse_confidence_based.py │ │ │ │ │ │ │ │ ├── densepose_cse_uniform.py │ │ │ │ │ │ │ │ ├── densepose_uniform.py │ │ │ │ │ │ │ │ ├── mask_from_densepose.py │ │ │ │ │ │ │ │ └── prediction_to_gt.py │ │ │ │ │ │ │ ├── transform │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ └── image.py │ │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ │ └── video │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── frame_selector.py │ │ │ │ │ │ │ │ └── video_keyframe_dataset.py │ │ │ │ │ │ ├── engine │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ └── trainer.py │ │ │ │ │ │ ├── evaluation │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── d2_evaluator_adapter.py │ │ │ │ │ │ │ ├── densepose_coco_evaluation.py │ │ │ │ │ │ │ ├── evaluator.py │ │ │ │ │ │ │ ├── mesh_alignment_evaluator.py │ │ │ │ │ │ │ └── tensor_storage.py │ │ │ │ │ │ ├── modeling │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ │ ├── confidence.py │ │ │ │ │ │ │ ├── cse │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── embedder.py │ │ │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ │ │ ├── vertex_direct_embedder.py │ │ │ │ │ │ │ │ └── vertex_feature_embedder.py │ │ │ │ │ │ │ ├── densepose_checkpoint.py │ │ │ │ │ │ │ ├── filter.py │ │ │ │ │ │ │ ├── hrfpn.py │ │ │ │ │ │ │ ├── hrnet.py │ │ │ │ │ │ │ ├── inference.py │ │ │ │ │ │ │ ├── losses │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ │ ├── chart_with_confidences.py │ │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ │ ├── cycle_pix2shape.py │ │ │ │ │ │ │ │ ├── cycle_shape2shape.py │ │ │ │ │ │ │ │ ├── embed.py │ │ │ │ │ │ │ │ ├── embed_utils.py │ │ │ │ │ │ │ │ ├── mask.py │ │ │ │ │ │ │ │ ├── mask_or_segm.py │ │ │ │ │ │ │ │ ├── registry.py │ │ │ │ │ │ │ │ ├── segm.py │ │ │ │ │ │ │ │ ├── soft_embed.py │ │ │ │ │ │ │ │ └── utils.py │ │ │ │ │ │ │ ├── predictors │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ │ ├── chart_confidence.py │ │ │ │ │ │ │ │ ├── chart_with_confidence.py │ │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ │ ├── cse_confidence.py │ │ │ │ │ │ │ │ ├── cse_with_confidence.py │ │ │ │ │ │ │ │ └── registry.py │ │ │ │ │ │ │ ├── roi_heads │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── deeplab.py │ │ │ │ │ │ │ │ ├── registry.py │ │ │ │ │ │ │ │ ├── roi_head.py │ │ │ │ │ │ │ │ └── v1convx.py │ │ │ │ │ │ │ ├── test_time_augmentation.py │ │ │ │ │ │ │ └── utils.py │ │ │ │ │ │ ├── structures │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ ├── chart_confidence.py │ │ │ │ │ │ │ ├── chart_result.py │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ ├── cse_confidence.py │ │ │ │ │ │ │ ├── data_relative.py │ │ │ │ │ │ │ ├── list.py │ │ │ │ │ │ │ ├── mesh.py │ │ │ │ │ │ │ └── transform_data.py │ │ │ │ │ │ ├── utils │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── dbhelper.py │ │ │ │ │ │ │ ├── logger.py │ │ │ │ │ │ │ └── transform.py │ │ │ │ │ │ └── vis │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── base.py │ │ │ │ │ │ │ ├── bounding_box.py │ │ │ │ │ │ │ ├── densepose_data_points.py │ │ │ │ │ │ │ ├── densepose_outputs_iuv.py │ │ │ │ │ │ │ ├── densepose_outputs_vertex.py │ │ │ │ │ │ │ ├── densepose_results.py │ │ │ │ │ │ │ ├── densepose_results_textures.py │ │ │ │ │ │ │ └── extractor.py │ │ │ │ │ ├── dev │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── run_inference_tests.sh │ │ │ │ │ │ └── run_instant_tests.sh │ │ │ │ │ ├── doc │ │ │ │ │ │ ├── BOOTSTRAPPING_PIPELINE.md │ │ │ │ │ │ ├── DENSEPOSE_CSE.md │ │ │ │ │ │ ├── DENSEPOSE_DATASETS.md │ │ │ │ │ │ ├── DENSEPOSE_IUV.md │ │ │ │ │ │ ├── GETTING_STARTED.md │ │ │ │ │ │ ├── RELEASE_2020_04.md │ │ │ │ │ │ ├── RELEASE_2021_03.md │ │ │ │ │ │ ├── RELEASE_2021_06.md │ │ │ │ │ │ ├── TOOL_APPLY_NET.md │ │ │ │ │ │ └── TOOL_QUERY_DB.md │ │ │ │ │ ├── query_db.py │ │ │ │ │ ├── setup.py │ │ │ │ │ ├── tests │ │ │ │ │ │ ├── common.py │ │ │ │ │ │ ├── test_chart_based_annotations_accumulator.py │ │ │ │ │ │ ├── test_combine_data_loader.py │ │ │ │ │ │ ├── test_cse_annotations_accumulator.py │ │ │ │ │ │ ├── test_dataset_loaded_annotations.py │ │ │ │ │ │ ├── test_frame_selector.py │ │ │ │ │ │ ├── test_image_list_dataset.py │ │ │ │ │ │ ├── test_image_resize_transform.py │ │ │ │ │ │ ├── test_model_e2e.py │ │ │ │ │ │ ├── test_setup.py │ │ │ │ │ │ ├── test_structures.py │ │ │ │ │ │ ├── test_tensor_storage.py │ │ │ │ │ │ └── test_video_keyframe_dataset.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── MViTv2 │ │ │ │ │ ├── README.md │ │ │ │ │ └── configs │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_3x.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_3x.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_lsj_3x.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_lsj_50ep.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_s_3x.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_t_3x.py │ │ │ │ │ │ ├── common │ │ │ │ │ │ ├── coco_loader.py │ │ │ │ │ │ └── coco_loader_lsj.py │ │ │ │ │ │ └── mask_rcnn_mvitv2_t_3x.py │ │ │ │ ├── Panoptic-DeepLab │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── COCO-PanopticSegmentation │ │ │ │ │ │ │ └── panoptic_deeplab_R_52_os16_mg124_poly_200k_bs64_crop_640_640_coco_dsconv.yaml │ │ │ │ │ │ └── Cityscapes-PanopticSegmentation │ │ │ │ │ │ │ ├── Base-PanopticDeepLab-OS16.yaml │ │ │ │ │ │ │ ├── panoptic_deeplab_R_52_os16_mg124_poly_90k_bs32_crop_512_1024.yaml │ │ │ │ │ │ │ └── panoptic_deeplab_R_52_os16_mg124_poly_90k_bs32_crop_512_1024_dsconv.yaml │ │ │ │ │ ├── panoptic_deeplab │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ │ ├── panoptic_seg.py │ │ │ │ │ │ ├── post_processing.py │ │ │ │ │ │ └── target_generator.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── PointRend │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── InstanceSegmentation │ │ │ │ │ │ │ ├── Base-Implicit-PointRend.yaml │ │ │ │ │ │ │ ├── Base-PointRend-RCNN-FPN.yaml │ │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_1x_coco.yaml │ │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_3x_coco.yaml │ │ │ │ │ │ │ ├── pointrend_rcnn_R_101_FPN_3x_coco.yaml │ │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_1x_cityscapes.yaml │ │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_1x_coco.yaml │ │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_3x_coco.yaml │ │ │ │ │ │ │ └── pointrend_rcnn_X_101_32x8d_FPN_3x_coco.yaml │ │ │ │ │ │ └── SemanticSegmentation │ │ │ │ │ │ │ ├── Base-PointRend-Semantic-FPN.yaml │ │ │ │ │ │ │ └── pointrend_semantic_R_101_FPN_1x_cityscapes.yaml │ │ │ │ │ ├── point_rend │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── color_augmentation.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── mask_head.py │ │ │ │ │ │ ├── point_features.py │ │ │ │ │ │ ├── point_head.py │ │ │ │ │ │ ├── roi_heads.py │ │ │ │ │ │ └── semantic_seg.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── PointSup │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_3x_point_sup_point_aug_coco.yaml │ │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_point_sup_coco.yaml │ │ │ │ │ │ └── mask_rcnn_R_50_FPN_3x_point_sup_point_aug_coco.yaml │ │ │ │ │ ├── point_sup │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ │ ├── detection_utils.py │ │ │ │ │ │ ├── mask_head.py │ │ │ │ │ │ ├── point_utils.py │ │ │ │ │ │ └── register_point_annotations.py │ │ │ │ │ ├── tools │ │ │ │ │ │ └── prepare_coco_point_annotations_without_masks.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── README.md │ │ │ │ ├── Rethinking-BatchNorm │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── mask_rcnn_BNhead.py │ │ │ │ │ │ ├── mask_rcnn_BNhead_batch_stats.py │ │ │ │ │ │ ├── mask_rcnn_BNhead_shuffle.py │ │ │ │ │ │ ├── mask_rcnn_SyncBNhead.py │ │ │ │ │ │ ├── retinanet_SyncBNhead.py │ │ │ │ │ │ └── retinanet_SyncBNhead_SharedTraining.py │ │ │ │ │ └── retinanet-eval-domain-specific.py │ │ │ │ ├── TensorMask │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── Base-TensorMask.yaml │ │ │ │ │ │ ├── tensormask_R_50_FPN_1x.yaml │ │ │ │ │ │ └── tensormask_R_50_FPN_6x.yaml │ │ │ │ │ ├── setup.py │ │ │ │ │ ├── tensormask │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── arch.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ └── layers │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── csrc │ │ │ │ │ │ │ ├── SwapAlign2Nat │ │ │ │ │ │ │ │ ├── SwapAlign2Nat.h │ │ │ │ │ │ │ │ └── SwapAlign2Nat_cuda.cu │ │ │ │ │ │ │ └── vision.cpp │ │ │ │ │ │ │ └── swap_align2nat.py │ │ │ │ │ ├── tests │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── test_swap_align2nat.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── TridentNet │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── Base-TridentNet-Fast-C4.yaml │ │ │ │ │ │ ├── tridentnet_fast_R_101_C4_3x.yaml │ │ │ │ │ │ ├── tridentnet_fast_R_50_C4_1x.yaml │ │ │ │ │ │ └── tridentnet_fast_R_50_C4_3x.yaml │ │ │ │ │ ├── train_net.py │ │ │ │ │ └── tridentnet │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── trident_backbone.py │ │ │ │ │ │ ├── trident_conv.py │ │ │ │ │ │ ├── trident_rcnn.py │ │ │ │ │ │ └── trident_rpn.py │ │ │ │ └── ViTDet │ │ │ │ │ ├── README.md │ │ │ │ │ └── configs │ │ │ │ │ ├── COCO │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_36ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_swin_b_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_swin_l_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vimdet_b_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vimdet_s_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vimdet_t_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vimdet_t_100ep_adj1.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_b_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_h_75ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_l_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_s_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_t_100ep.py │ │ │ │ │ ├── lite_cascade_mask_rcnn_vimdet_t_100ep.py │ │ │ │ │ ├── mask_rcnn_vimdet_b_100ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_b_100ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_h_75ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_l_100ep.py │ │ │ │ │ └── mask_rcnn_vitdet_t_100ep.py │ │ │ │ │ ├── LVIS │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_swin_b_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_swin_l_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_b_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_h_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_l_100ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_b_100ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_h_100ep.py │ │ │ │ │ └── mask_rcnn_vitdet_l_100ep.py │ │ │ │ │ └── common │ │ │ │ │ └── coco_loader_lsj.py │ │ │ ├── scripts │ │ │ │ ├── eval_vim_tiny_vimdet.sh │ │ │ │ └── ft_vim_tiny_vimdet.sh │ │ │ ├── setup.cfg │ │ │ ├── setup.py │ │ │ ├── tests │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── config │ │ │ │ │ ├── dir1 │ │ │ │ │ │ ├── bad_import.py │ │ │ │ │ │ ├── bad_import2.py │ │ │ │ │ │ ├── dir1_a.py │ │ │ │ │ │ ├── dir1_b.py │ │ │ │ │ │ └── load_rel.py │ │ │ │ │ ├── root_cfg.py │ │ │ │ │ ├── test_instantiate_config.py │ │ │ │ │ ├── test_lazy_config.py │ │ │ │ │ └── test_yacs_config.py │ │ │ │ ├── data │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_coco.py │ │ │ │ │ ├── test_coco_evaluation.py │ │ │ │ │ ├── test_dataset.py │ │ │ │ │ ├── test_detection_utils.py │ │ │ │ │ ├── test_rotation_transform.py │ │ │ │ │ ├── test_sampler.py │ │ │ │ │ └── test_transforms.py │ │ │ │ ├── export │ │ │ │ │ └── test_c10.py │ │ │ │ ├── layers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_blocks.py │ │ │ │ │ ├── test_deformable.py │ │ │ │ │ ├── test_losses.py │ │ │ │ │ ├── test_mask_ops.py │ │ │ │ │ ├── test_nms.py │ │ │ │ │ ├── test_nms_rotated.py │ │ │ │ │ ├── test_roi_align.py │ │ │ │ │ └── test_roi_align_rotated.py │ │ │ │ ├── modeling │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_anchor_generator.py │ │ │ │ │ ├── test_backbone.py │ │ │ │ │ ├── test_box2box_transform.py │ │ │ │ │ ├── test_fast_rcnn.py │ │ │ │ │ ├── test_matcher.py │ │ │ │ │ ├── test_mmdet.py │ │ │ │ │ ├── test_model_e2e.py │ │ │ │ │ ├── test_roi_heads.py │ │ │ │ │ ├── test_roi_pooler.py │ │ │ │ │ └── test_rpn.py │ │ │ │ ├── structures │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_boxes.py │ │ │ │ │ ├── test_imagelist.py │ │ │ │ │ ├── test_instances.py │ │ │ │ │ ├── test_keypoints.py │ │ │ │ │ ├── test_masks.py │ │ │ │ │ └── test_rotated_boxes.py │ │ │ │ ├── test_checkpoint.py │ │ │ │ ├── test_engine.py │ │ │ │ ├── test_events.py │ │ │ │ ├── test_export_caffe2.py │ │ │ │ ├── test_export_onnx.py │ │ │ │ ├── test_export_torchscript.py │ │ │ │ ├── test_model_analysis.py │ │ │ │ ├── test_model_zoo.py │ │ │ │ ├── test_packaging.py │ │ │ │ ├── test_registry.py │ │ │ │ ├── test_scheduler.py │ │ │ │ ├── test_solver.py │ │ │ │ ├── test_visualizer.py │ │ │ │ ├── tracking │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_bbox_iou_tracker.py │ │ │ │ │ ├── test_hungarian_tracker.py │ │ │ │ │ ├── test_iou_weighted_hungarian_bbox_iou_tracker.py │ │ │ │ │ └── test_vanilla_hungarian_bbox_iou_tracker.py │ │ │ │ └── utils │ │ │ │ │ └── test_tensorboardx.py │ │ │ └── tools │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── analyze_model.py │ │ │ │ ├── benchmark.py │ │ │ │ ├── convert-torchvision-to-d2.py │ │ │ │ ├── deploy │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── README.md │ │ │ │ ├── export_model.py │ │ │ │ └── torchscript_mask_rcnn.cpp │ │ │ │ ├── lazyconfig_train_net.py │ │ │ │ ├── lightning_train_net.py │ │ │ │ ├── plain_train_net.py │ │ │ │ ├── train_net.py │ │ │ │ ├── visualize_data.py │ │ │ │ └── visualize_json_results.py │ │ ├── mamba-1p1p1 │ │ │ ├── .github │ │ │ │ └── workflows │ │ │ │ │ └── publish.yaml │ │ │ ├── .gitignore │ │ │ ├── .gitmodules │ │ │ ├── AUTHORS │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── assets │ │ │ │ └── selection.png │ │ │ ├── benchmarks │ │ │ │ └── benchmark_generation_mamba_simple.py │ │ │ ├── csrc │ │ │ │ └── selective_scan │ │ │ │ │ ├── reverse_scan.cuh │ │ │ │ │ ├── selective_scan.cpp │ │ │ │ │ ├── selective_scan.h │ │ │ │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ │ │ │ ├── selective_scan_bwd_bf16_real.cu │ │ │ │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ │ │ │ ├── selective_scan_bwd_fp16_real.cu │ │ │ │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ │ │ │ ├── selective_scan_bwd_fp32_real.cu │ │ │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ │ │ ├── selective_scan_common.h │ │ │ │ │ ├── selective_scan_fwd_bf16.cu │ │ │ │ │ ├── selective_scan_fwd_fp16.cu │ │ │ │ │ ├── selective_scan_fwd_fp32.cu │ │ │ │ │ ├── selective_scan_fwd_kernel.cuh │ │ │ │ │ ├── static_switch.h │ │ │ │ │ └── uninitialized_copy.cuh │ │ │ ├── evals │ │ │ │ └── lm_harness_eval.py │ │ │ ├── mamba_ssm │ │ │ │ ├── __init__.py │ │ │ │ ├── models │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config_mamba.py │ │ │ │ │ └── mixer_seq_simple.py │ │ │ │ ├── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── mamba_simple.py │ │ │ │ ├── ops │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── selective_scan_interface.py │ │ │ │ │ └── triton │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── layernorm.py │ │ │ │ │ │ └── selective_state_update.py │ │ │ │ └── utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── generation.py │ │ │ │ │ └── hf.py │ │ │ ├── setup.py │ │ │ └── tests │ │ │ │ └── ops │ │ │ │ ├── test_selective_scan.py │ │ │ │ └── triton │ │ │ │ └── test_selective_state_update.py │ │ ├── seg │ │ │ ├── README.md │ │ │ ├── backbone │ │ │ │ ├── __init__.py │ │ │ │ └── vim.py │ │ │ ├── configs │ │ │ │ ├── _base_ │ │ │ │ │ ├── datasets │ │ │ │ │ │ ├── ade20k.py │ │ │ │ │ │ ├── ade20k_1024x1024.py │ │ │ │ │ │ ├── ade20k_640x640.py │ │ │ │ │ │ └── ade20k_ms_eval.py │ │ │ │ │ ├── default_runtime.py │ │ │ │ │ ├── models │ │ │ │ │ │ ├── plainseg_vim.py │ │ │ │ │ │ ├── upernet_beit.py │ │ │ │ │ │ └── upernet_vim.py │ │ │ │ │ └── schedules │ │ │ │ │ │ ├── schedule_60k.py │ │ │ │ │ │ └── schedule_80k.py │ │ │ │ └── vim │ │ │ │ │ └── upernet │ │ │ │ │ ├── upernet_vim_small_24_512_slide_60k.py │ │ │ │ │ └── upernet_vim_tiny_24_512_slide_60k.py │ │ │ ├── mmcv_custom │ │ │ │ ├── __init__.py │ │ │ │ ├── apex_runner │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── apex_iter_based_runner.py │ │ │ │ │ ├── checkpoint.py │ │ │ │ │ └── optimizer.py │ │ │ │ ├── checkpoint.py │ │ │ │ ├── layer_decay_optimizer_constructor.py │ │ │ │ ├── resize_transform.py │ │ │ │ └── train_api.py │ │ │ ├── scripts │ │ │ │ ├── eval_vim_small_upernet.sh │ │ │ │ ├── eval_vim_tiny_upernet.sh │ │ │ │ ├── ft_vim_small_upernet.sh │ │ │ │ └── ft_vim_tiny_upernet.sh │ │ │ ├── seg-requirements.txt │ │ │ ├── test.py │ │ │ ├── tools │ │ │ │ └── analysis_tools │ │ │ │ │ ├── benchmark.py │ │ │ │ │ ├── benchmark_scripts.sh │ │ │ │ │ ├── deit-t-upernet.txt │ │ │ │ │ ├── plot_curve.ipynb │ │ │ │ │ └── vim-t-upernet.txt │ │ │ └── train.py │ │ └── vim │ │ │ ├── .gitignore │ │ │ ├── =0.9.0 │ │ │ ├── =1.8.0 │ │ │ ├── LICENSE │ │ │ ├── augment.py │ │ │ ├── calc_flops.py │ │ │ ├── datasets.py │ │ │ ├── engine.py │ │ │ ├── hubconf.py │ │ │ ├── losses.py │ │ │ ├── main.py │ │ │ ├── models │ │ │ ├── models_mamba.py │ │ │ ├── videomamba.py │ │ │ └── vim.py │ │ │ ├── rope.py │ │ │ ├── run_with_submitit.py │ │ │ ├── samplers.py │ │ │ ├── scripts │ │ │ └── ft-vim-t.sh │ │ │ ├── utils.py │ │ │ ├── vim_requirements.txt │ │ │ └── visualization.ipynb │ └── Predictor-delta │ │ ├── .gitignore │ │ ├── =1.1.0 │ │ ├── LICENSE.txt │ │ ├── README.md │ │ ├── assets │ │ ├── vim_pipeline_v1.9.png │ │ └── vim_teaser_v1.7.png │ │ ├── det │ │ ├── .circleci │ │ │ ├── config.yml │ │ │ └── import-tests.sh │ │ ├── .clang-format │ │ ├── .flake8 │ │ ├── .gitignore │ │ ├── GETTING_STARTED.md │ │ ├── INSTALL.md │ │ ├── LICENSE │ │ ├── MODEL_ZOO.md │ │ ├── README.md │ │ ├── configs │ │ │ ├── Base-RCNN-C4.yaml │ │ │ ├── Base-RCNN-DilatedC5.yaml │ │ │ ├── Base-RCNN-FPN.yaml │ │ │ ├── Base-RetinaNet.yaml │ │ │ ├── COCO-Detection │ │ │ │ ├── fast_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── faster_rcnn_R_101_C4_3x.yaml │ │ │ │ ├── faster_rcnn_R_101_DC5_3x.yaml │ │ │ │ ├── faster_rcnn_R_101_FPN_3x.yaml │ │ │ │ ├── faster_rcnn_R_50_C4_1x.yaml │ │ │ │ ├── faster_rcnn_R_50_C4_3x.yaml │ │ │ │ ├── faster_rcnn_R_50_DC5_1x.yaml │ │ │ │ ├── faster_rcnn_R_50_DC5_3x.yaml │ │ │ │ ├── faster_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── faster_rcnn_R_50_FPN_3x.yaml │ │ │ │ ├── faster_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ ├── fcos_R_50_FPN_1x.py │ │ │ │ ├── retinanet_R_101_FPN_3x.yaml │ │ │ │ ├── retinanet_R_50_FPN_1x.py │ │ │ │ ├── retinanet_R_50_FPN_1x.yaml │ │ │ │ ├── retinanet_R_50_FPN_3x.yaml │ │ │ │ ├── rpn_R_50_C4_1x.yaml │ │ │ │ └── rpn_R_50_FPN_1x.yaml │ │ │ ├── COCO-InstanceSegmentation │ │ │ │ ├── mask_rcnn_R_101_C4_3x.yaml │ │ │ │ ├── mask_rcnn_R_101_DC5_3x.yaml │ │ │ │ ├── mask_rcnn_R_101_FPN_3x.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_1x.py │ │ │ │ ├── mask_rcnn_R_50_C4_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_3x.yaml │ │ │ │ ├── mask_rcnn_R_50_DC5_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_DC5_3x.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.py │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_giou.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_3x.yaml │ │ │ │ ├── mask_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_fpn_1x.py │ │ │ │ └── mask_rcnn_regnety_4gf_dds_fpn_1x.py │ │ │ ├── COCO-Keypoints │ │ │ │ ├── Base-Keypoint-RCNN-FPN.yaml │ │ │ │ ├── keypoint_rcnn_R_101_FPN_3x.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.py │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_3x.yaml │ │ │ │ └── keypoint_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ ├── COCO-PanopticSegmentation │ │ │ │ ├── Base-Panoptic-FPN.yaml │ │ │ │ ├── panoptic_fpn_R_101_3x.yaml │ │ │ │ ├── panoptic_fpn_R_50_1x.py │ │ │ │ ├── panoptic_fpn_R_50_1x.yaml │ │ │ │ └── panoptic_fpn_R_50_3x.yaml │ │ │ ├── Cityscapes │ │ │ │ └── mask_rcnn_R_50_FPN.yaml │ │ │ ├── Detectron1-Comparisons │ │ │ │ ├── README.md │ │ │ │ ├── faster_rcnn_R_50_FPN_noaug_1x.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.yaml │ │ │ │ └── mask_rcnn_R_50_FPN_noaug_1x.yaml │ │ │ ├── LVISv0.5-InstanceSegmentation │ │ │ │ ├── mask_rcnn_R_101_FPN_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ └── mask_rcnn_X_101_32x8d_FPN_1x.yaml │ │ │ ├── LVISv1-InstanceSegmentation │ │ │ │ ├── mask_rcnn_R_101_FPN_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ └── mask_rcnn_X_101_32x8d_FPN_1x.yaml │ │ │ ├── Misc │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_3x.yaml │ │ │ │ ├── cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_cls_agnostic.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_dconv_c3-c5.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_dconv_c3-c5.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_gn.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_syncbn.yaml │ │ │ │ ├── mmdet_mask_rcnn_R_50_FPN_1x.py │ │ │ │ ├── panoptic_fpn_R_101_dconv_cascade_gn_3x.yaml │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_3x_gn.yaml │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_9x_gn.yaml │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_9x_syncbn.yaml │ │ │ │ ├── semantic_R_50_FPN_1x.yaml │ │ │ │ └── torchvision_imagenet_R_50.py │ │ │ ├── PascalVOC-Detection │ │ │ │ ├── faster_rcnn_R_50_C4.yaml │ │ │ │ └── faster_rcnn_R_50_FPN.yaml │ │ │ ├── common │ │ │ │ ├── README.md │ │ │ │ ├── coco_schedule.py │ │ │ │ ├── data │ │ │ │ │ ├── coco.py │ │ │ │ │ ├── coco_keypoint.py │ │ │ │ │ ├── coco_panoptic_separated.py │ │ │ │ │ └── constants.py │ │ │ │ ├── models │ │ │ │ │ ├── cascade_rcnn.py │ │ │ │ │ ├── fcos.py │ │ │ │ │ ├── keypoint_rcnn_fpn.py │ │ │ │ │ ├── mask_rcnn_c4.py │ │ │ │ │ ├── mask_rcnn_fpn.py │ │ │ │ │ ├── mask_rcnn_vimdet.py │ │ │ │ │ ├── mask_rcnn_vitdet.py │ │ │ │ │ ├── panoptic_fpn.py │ │ │ │ │ └── retinanet.py │ │ │ │ ├── optim.py │ │ │ │ └── train.py │ │ │ ├── new_baselines │ │ │ │ ├── mask_rcnn_R_101_FPN_100ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_101_FPN_200ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_101_FPN_400ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_50_FPN_100ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_50_FPN_200ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_50_FPN_400ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_50_FPN_50ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_100ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_200ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_400ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnety_4gf_dds_FPN_100ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnety_4gf_dds_FPN_200ep_LSJ.py │ │ │ │ └── mask_rcnn_regnety_4gf_dds_FPN_400ep_LSJ.py │ │ │ └── quick_schedules │ │ │ │ ├── README.md │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ ├── fast_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── fast_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_normalized_training_acc_test.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_GCV_instant_test.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_inference_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_instant_test.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_training_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_DC5_inference_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_pred_boxes_training_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ ├── panoptic_fpn_R_50_inference_acc_test.yaml │ │ │ │ ├── panoptic_fpn_R_50_instant_test.yaml │ │ │ │ ├── panoptic_fpn_R_50_training_acc_test.yaml │ │ │ │ ├── retinanet_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── retinanet_R_50_FPN_instant_test.yaml │ │ │ │ ├── rpn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── rpn_R_50_FPN_instant_test.yaml │ │ │ │ ├── semantic_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── semantic_R_50_FPN_instant_test.yaml │ │ │ │ └── semantic_R_50_FPN_training_acc_test.yaml │ │ ├── datasets │ │ │ ├── README.md │ │ │ ├── prepare_ade20k_sem_seg.py │ │ │ ├── prepare_cocofied_lvis.py │ │ │ ├── prepare_for_tests.sh │ │ │ └── prepare_panoptic_fpn.py │ │ ├── demo │ │ │ ├── README.md │ │ │ ├── demo.py │ │ │ └── predictor.py │ │ ├── det-requirements.txt │ │ ├── detectron2 │ │ │ ├── __init__.py │ │ │ ├── checkpoint │ │ │ │ ├── __init__.py │ │ │ │ ├── c2_model_loading.py │ │ │ │ ├── catalog.py │ │ │ │ └── detection_checkpoint.py │ │ │ ├── config │ │ │ │ ├── __init__.py │ │ │ │ ├── compat.py │ │ │ │ ├── config.py │ │ │ │ ├── defaults.py │ │ │ │ ├── instantiate.py │ │ │ │ └── lazy.py │ │ │ ├── data │ │ │ │ ├── __init__.py │ │ │ │ ├── benchmark.py │ │ │ │ ├── build.py │ │ │ │ ├── catalog.py │ │ │ │ ├── common.py │ │ │ │ ├── dataset_mapper.py │ │ │ │ ├── datasets │ │ │ │ │ ├── README.md │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── builtin.py │ │ │ │ │ ├── builtin_meta.py │ │ │ │ │ ├── cityscapes.py │ │ │ │ │ ├── cityscapes_panoptic.py │ │ │ │ │ ├── coco.py │ │ │ │ │ ├── coco_panoptic.py │ │ │ │ │ ├── lvis.py │ │ │ │ │ ├── lvis_v0_5_categories.py │ │ │ │ │ ├── lvis_v1_categories.py │ │ │ │ │ ├── lvis_v1_category_image_count.py │ │ │ │ │ ├── pascal_voc.py │ │ │ │ │ └── register_coco.py │ │ │ │ ├── detection_utils.py │ │ │ │ ├── samplers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── distributed_sampler.py │ │ │ │ │ └── grouped_batch_sampler.py │ │ │ │ └── transforms │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── augmentation.py │ │ │ │ │ ├── augmentation_impl.py │ │ │ │ │ └── transform.py │ │ │ ├── engine │ │ │ │ ├── __init__.py │ │ │ │ ├── defaults.py │ │ │ │ ├── hooks.py │ │ │ │ ├── launch.py │ │ │ │ └── train_loop.py │ │ │ ├── evaluation │ │ │ │ ├── __init__.py │ │ │ │ ├── cityscapes_evaluation.py │ │ │ │ ├── coco_evaluation.py │ │ │ │ ├── evaluator.py │ │ │ │ ├── fast_eval_api.py │ │ │ │ ├── lvis_evaluation.py │ │ │ │ ├── panoptic_evaluation.py │ │ │ │ ├── pascal_voc_evaluation.py │ │ │ │ ├── rotated_coco_evaluation.py │ │ │ │ ├── sem_seg_evaluation.py │ │ │ │ └── testing.py │ │ │ ├── export │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── api.py │ │ │ │ ├── c10.py │ │ │ │ ├── caffe2_export.py │ │ │ │ ├── caffe2_inference.py │ │ │ │ ├── caffe2_modeling.py │ │ │ │ ├── caffe2_patch.py │ │ │ │ ├── flatten.py │ │ │ │ ├── shared.py │ │ │ │ ├── torchscript.py │ │ │ │ └── torchscript_patch.py │ │ │ ├── layers │ │ │ │ ├── __init__.py │ │ │ │ ├── aspp.py │ │ │ │ ├── batch_norm.py │ │ │ │ ├── blocks.py │ │ │ │ ├── csrc │ │ │ │ │ ├── README.md │ │ │ │ │ ├── ROIAlignRotated │ │ │ │ │ │ ├── ROIAlignRotated.h │ │ │ │ │ │ ├── ROIAlignRotated_cpu.cpp │ │ │ │ │ │ └── ROIAlignRotated_cuda.cu │ │ │ │ │ ├── box_iou_rotated │ │ │ │ │ │ ├── box_iou_rotated.h │ │ │ │ │ │ ├── box_iou_rotated_cpu.cpp │ │ │ │ │ │ ├── box_iou_rotated_cuda.cu │ │ │ │ │ │ └── box_iou_rotated_utils.h │ │ │ │ │ ├── cocoeval │ │ │ │ │ │ ├── cocoeval.cpp │ │ │ │ │ │ └── cocoeval.h │ │ │ │ │ ├── cuda_version.cu │ │ │ │ │ ├── deformable │ │ │ │ │ │ ├── deform_conv.h │ │ │ │ │ │ ├── deform_conv_cuda.cu │ │ │ │ │ │ └── deform_conv_cuda_kernel.cu │ │ │ │ │ ├── nms_rotated │ │ │ │ │ │ ├── nms_rotated.h │ │ │ │ │ │ ├── nms_rotated_cpu.cpp │ │ │ │ │ │ └── nms_rotated_cuda.cu │ │ │ │ │ └── vision.cpp │ │ │ │ ├── deform_conv.py │ │ │ │ ├── losses.py │ │ │ │ ├── mask_ops.py │ │ │ │ ├── nms.py │ │ │ │ ├── roi_align.py │ │ │ │ ├── roi_align_rotated.py │ │ │ │ ├── rotated_boxes.py │ │ │ │ ├── shape_spec.py │ │ │ │ └── wrappers.py │ │ │ ├── model_zoo │ │ │ │ ├── __init__.py │ │ │ │ └── model_zoo.py │ │ │ ├── modeling │ │ │ │ ├── __init__.py │ │ │ │ ├── anchor_generator.py │ │ │ │ ├── backbone │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── backbone.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── fpn.py │ │ │ │ │ ├── mvit.py │ │ │ │ │ ├── regnet.py │ │ │ │ │ ├── resnet.py │ │ │ │ │ ├── swin.py │ │ │ │ │ ├── utils.py │ │ │ │ │ ├── vim.py │ │ │ │ │ └── vit.py │ │ │ │ ├── box_regression.py │ │ │ │ ├── matcher.py │ │ │ │ ├── meta_arch │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── dense_detector.py │ │ │ │ │ ├── fcos.py │ │ │ │ │ ├── panoptic_fpn.py │ │ │ │ │ ├── rcnn.py │ │ │ │ │ ├── retinanet.py │ │ │ │ │ └── semantic_seg.py │ │ │ │ ├── mmdet_wrapper.py │ │ │ │ ├── poolers.py │ │ │ │ ├── postprocessing.py │ │ │ │ ├── proposal_generator │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── proposal_utils.py │ │ │ │ │ ├── rpn.py │ │ │ │ │ └── rrpn.py │ │ │ │ ├── roi_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── box_head.py │ │ │ │ │ ├── cascade_rcnn.py │ │ │ │ │ ├── fast_rcnn.py │ │ │ │ │ ├── keypoint_head.py │ │ │ │ │ ├── mask_head.py │ │ │ │ │ ├── roi_heads.py │ │ │ │ │ └── rotated_fast_rcnn.py │ │ │ │ ├── sampling.py │ │ │ │ └── test_time_augmentation.py │ │ │ ├── projects │ │ │ │ ├── README.md │ │ │ │ └── __init__.py │ │ │ ├── solver │ │ │ │ ├── __init__.py │ │ │ │ ├── build.py │ │ │ │ └── lr_scheduler.py │ │ │ ├── structures │ │ │ │ ├── __init__.py │ │ │ │ ├── boxes.py │ │ │ │ ├── image_list.py │ │ │ │ ├── instances.py │ │ │ │ ├── keypoints.py │ │ │ │ ├── masks.py │ │ │ │ └── rotated_boxes.py │ │ │ ├── tracking │ │ │ │ ├── __init__.py │ │ │ │ ├── base_tracker.py │ │ │ │ ├── bbox_iou_tracker.py │ │ │ │ ├── hungarian_tracker.py │ │ │ │ ├── iou_weighted_hungarian_bbox_iou_tracker.py │ │ │ │ ├── utils.py │ │ │ │ └── vanilla_hungarian_bbox_iou_tracker.py │ │ │ └── utils │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── analysis.py │ │ │ │ ├── collect_env.py │ │ │ │ ├── colormap.py │ │ │ │ ├── comm.py │ │ │ │ ├── develop.py │ │ │ │ ├── env.py │ │ │ │ ├── events.py │ │ │ │ ├── file_io.py │ │ │ │ ├── logger.py │ │ │ │ ├── memory.py │ │ │ │ ├── registry.py │ │ │ │ ├── serialize.py │ │ │ │ ├── testing.py │ │ │ │ ├── tracing.py │ │ │ │ ├── video_visualizer.py │ │ │ │ └── visualizer.py │ │ ├── dev │ │ │ ├── README.md │ │ │ ├── linter.sh │ │ │ ├── packaging │ │ │ │ ├── README.md │ │ │ │ ├── build_all_wheels.sh │ │ │ │ ├── build_wheel.sh │ │ │ │ ├── gen_install_table.py │ │ │ │ ├── gen_wheel_index.sh │ │ │ │ └── pkg_helpers.bash │ │ │ ├── parse_results.sh │ │ │ ├── run_inference_tests.sh │ │ │ └── run_instant_tests.sh │ │ ├── docker │ │ │ ├── Dockerfile │ │ │ ├── README.md │ │ │ ├── deploy.Dockerfile │ │ │ └── docker-compose.yml │ │ ├── docs │ │ │ ├── .gitignore │ │ │ ├── Makefile │ │ │ ├── README.md │ │ │ ├── _static │ │ │ │ └── css │ │ │ │ │ └── custom.css │ │ │ ├── conf.py │ │ │ ├── index.rst │ │ │ ├── modules │ │ │ │ ├── checkpoint.rst │ │ │ │ ├── config.rst │ │ │ │ ├── data.rst │ │ │ │ ├── data_transforms.rst │ │ │ │ ├── engine.rst │ │ │ │ ├── evaluation.rst │ │ │ │ ├── export.rst │ │ │ │ ├── fvcore.rst │ │ │ │ ├── index.rst │ │ │ │ ├── layers.rst │ │ │ │ ├── model_zoo.rst │ │ │ │ ├── modeling.rst │ │ │ │ ├── solver.rst │ │ │ │ ├── structures.rst │ │ │ │ └── utils.rst │ │ │ ├── notes │ │ │ │ ├── benchmarks.md │ │ │ │ ├── changelog.md │ │ │ │ ├── compatibility.md │ │ │ │ ├── contributing.md │ │ │ │ └── index.rst │ │ │ ├── requirements.txt │ │ │ └── tutorials │ │ │ │ ├── README.md │ │ │ │ ├── augmentation.md │ │ │ │ ├── builtin_datasets.md │ │ │ │ ├── configs.md │ │ │ │ ├── data_loading.md │ │ │ │ ├── datasets.md │ │ │ │ ├── deployment.md │ │ │ │ ├── evaluation.md │ │ │ │ ├── extend.md │ │ │ │ ├── getting_started.md │ │ │ │ ├── index.rst │ │ │ │ ├── install.md │ │ │ │ ├── lazyconfigs.md │ │ │ │ ├── models.md │ │ │ │ ├── training.md │ │ │ │ └── write-models.md │ │ ├── projects │ │ │ ├── DeepLab │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ └── Cityscapes-SemanticSegmentation │ │ │ │ │ │ ├── Base-DeepLabV3-OS16-Semantic.yaml │ │ │ │ │ │ ├── deeplab_v3_R_103_os16_mg124_poly_90k_bs16.yaml │ │ │ │ │ │ └── deeplab_v3_plus_R_103_os16_mg124_poly_90k_bs16.yaml │ │ │ │ ├── deeplab │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build_solver.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── loss.py │ │ │ │ │ ├── lr_scheduler.py │ │ │ │ │ ├── resnet.py │ │ │ │ │ └── semantic_seg.py │ │ │ │ └── train_net.py │ │ │ ├── DensePose │ │ │ │ ├── README.md │ │ │ │ ├── apply_net.py │ │ │ │ ├── configs │ │ │ │ │ ├── Base-DensePose-RCNN-FPN.yaml │ │ │ │ │ ├── HRNet │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w32_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w40_s1x.yaml │ │ │ │ │ │ └── densepose_rcnn_HRFPN_HRNet_w48_s1x.yaml │ │ │ │ │ ├── cse │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN-Human.yaml │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_soft_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_soft_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_soft_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_4k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_i2m_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_m2m_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_4k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_maskonly_24k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_chimps_finetune_4k.yaml │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_soft_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC1M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC1_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC2M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC2_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC1M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC1_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC2M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC2_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x_legacy.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC1M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC1_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC2M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC2_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x_legacy.yaml │ │ │ │ │ ├── evolution │ │ │ │ │ │ ├── Base-RCNN-FPN-Atop10P_CA.yaml │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA.yaml │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_coarsesegm.yaml │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_finesegm.yaml │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_uniform.yaml │ │ │ │ │ │ └── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_uv.yaml │ │ │ │ │ └── quick_schedules │ │ │ │ │ │ ├── cse │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_instant_test.yaml │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_soft_animals_finetune_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w32_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_TTA_inference_acc_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ ├── densepose │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── converters │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── base.py │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ ├── chart_output_hflip.py │ │ │ │ │ │ ├── chart_output_to_chart_result.py │ │ │ │ │ │ ├── hflip.py │ │ │ │ │ │ ├── segm_to_mask.py │ │ │ │ │ │ ├── to_chart_result.py │ │ │ │ │ │ └── to_mask.py │ │ │ │ │ ├── data │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── combined_loader.py │ │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ │ ├── datasets │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ ├── chimpnsee.py │ │ │ │ │ │ │ ├── coco.py │ │ │ │ │ │ │ ├── dataset_type.py │ │ │ │ │ │ │ └── lvis.py │ │ │ │ │ │ ├── image_list_dataset.py │ │ │ │ │ │ ├── inference_based_loader.py │ │ │ │ │ │ ├── meshes │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ └── catalog.py │ │ │ │ │ │ ├── samplers │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── densepose_base.py │ │ │ │ │ │ │ ├── densepose_confidence_based.py │ │ │ │ │ │ │ ├── densepose_cse_base.py │ │ │ │ │ │ │ ├── densepose_cse_confidence_based.py │ │ │ │ │ │ │ ├── densepose_cse_uniform.py │ │ │ │ │ │ │ ├── densepose_uniform.py │ │ │ │ │ │ │ ├── mask_from_densepose.py │ │ │ │ │ │ │ └── prediction_to_gt.py │ │ │ │ │ │ ├── transform │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ └── image.py │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ └── video │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── frame_selector.py │ │ │ │ │ │ │ └── video_keyframe_dataset.py │ │ │ │ │ ├── engine │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── trainer.py │ │ │ │ │ ├── evaluation │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── d2_evaluator_adapter.py │ │ │ │ │ │ ├── densepose_coco_evaluation.py │ │ │ │ │ │ ├── evaluator.py │ │ │ │ │ │ ├── mesh_alignment_evaluator.py │ │ │ │ │ │ └── tensor_storage.py │ │ │ │ │ ├── modeling │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── confidence.py │ │ │ │ │ │ ├── cse │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── embedder.py │ │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ │ ├── vertex_direct_embedder.py │ │ │ │ │ │ │ └── vertex_feature_embedder.py │ │ │ │ │ │ ├── densepose_checkpoint.py │ │ │ │ │ │ ├── filter.py │ │ │ │ │ │ ├── hrfpn.py │ │ │ │ │ │ ├── hrnet.py │ │ │ │ │ │ ├── inference.py │ │ │ │ │ │ ├── losses │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ ├── chart_with_confidences.py │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ ├── cycle_pix2shape.py │ │ │ │ │ │ │ ├── cycle_shape2shape.py │ │ │ │ │ │ │ ├── embed.py │ │ │ │ │ │ │ ├── embed_utils.py │ │ │ │ │ │ │ ├── mask.py │ │ │ │ │ │ │ ├── mask_or_segm.py │ │ │ │ │ │ │ ├── registry.py │ │ │ │ │ │ │ ├── segm.py │ │ │ │ │ │ │ ├── soft_embed.py │ │ │ │ │ │ │ └── utils.py │ │ │ │ │ │ ├── predictors │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ ├── chart_confidence.py │ │ │ │ │ │ │ ├── chart_with_confidence.py │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ ├── cse_confidence.py │ │ │ │ │ │ │ ├── cse_with_confidence.py │ │ │ │ │ │ │ └── registry.py │ │ │ │ │ │ ├── roi_heads │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── deeplab.py │ │ │ │ │ │ │ ├── registry.py │ │ │ │ │ │ │ ├── roi_head.py │ │ │ │ │ │ │ └── v1convx.py │ │ │ │ │ │ ├── test_time_augmentation.py │ │ │ │ │ │ └── utils.py │ │ │ │ │ ├── structures │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ ├── chart_confidence.py │ │ │ │ │ │ ├── chart_result.py │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ ├── cse_confidence.py │ │ │ │ │ │ ├── data_relative.py │ │ │ │ │ │ ├── list.py │ │ │ │ │ │ ├── mesh.py │ │ │ │ │ │ └── transform_data.py │ │ │ │ │ ├── utils │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── dbhelper.py │ │ │ │ │ │ ├── logger.py │ │ │ │ │ │ └── transform.py │ │ │ │ │ └── vis │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── base.py │ │ │ │ │ │ ├── bounding_box.py │ │ │ │ │ │ ├── densepose_data_points.py │ │ │ │ │ │ ├── densepose_outputs_iuv.py │ │ │ │ │ │ ├── densepose_outputs_vertex.py │ │ │ │ │ │ ├── densepose_results.py │ │ │ │ │ │ ├── densepose_results_textures.py │ │ │ │ │ │ └── extractor.py │ │ │ │ ├── dev │ │ │ │ │ ├── README.md │ │ │ │ │ ├── run_inference_tests.sh │ │ │ │ │ └── run_instant_tests.sh │ │ │ │ ├── doc │ │ │ │ │ ├── BOOTSTRAPPING_PIPELINE.md │ │ │ │ │ ├── DENSEPOSE_CSE.md │ │ │ │ │ ├── DENSEPOSE_DATASETS.md │ │ │ │ │ ├── DENSEPOSE_IUV.md │ │ │ │ │ ├── GETTING_STARTED.md │ │ │ │ │ ├── RELEASE_2020_04.md │ │ │ │ │ ├── RELEASE_2021_03.md │ │ │ │ │ ├── RELEASE_2021_06.md │ │ │ │ │ ├── TOOL_APPLY_NET.md │ │ │ │ │ └── TOOL_QUERY_DB.md │ │ │ │ ├── query_db.py │ │ │ │ ├── setup.py │ │ │ │ ├── tests │ │ │ │ │ ├── common.py │ │ │ │ │ ├── test_chart_based_annotations_accumulator.py │ │ │ │ │ ├── test_combine_data_loader.py │ │ │ │ │ ├── test_cse_annotations_accumulator.py │ │ │ │ │ ├── test_dataset_loaded_annotations.py │ │ │ │ │ ├── test_frame_selector.py │ │ │ │ │ ├── test_image_list_dataset.py │ │ │ │ │ ├── test_image_resize_transform.py │ │ │ │ │ ├── test_model_e2e.py │ │ │ │ │ ├── test_setup.py │ │ │ │ │ ├── test_structures.py │ │ │ │ │ ├── test_tensor_storage.py │ │ │ │ │ └── test_video_keyframe_dataset.py │ │ │ │ └── train_net.py │ │ │ ├── MViTv2 │ │ │ │ ├── README.md │ │ │ │ └── configs │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_3x.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_3x.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_lsj_3x.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_lsj_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_s_3x.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_t_3x.py │ │ │ │ │ ├── common │ │ │ │ │ ├── coco_loader.py │ │ │ │ │ └── coco_loader_lsj.py │ │ │ │ │ └── mask_rcnn_mvitv2_t_3x.py │ │ │ ├── Panoptic-DeepLab │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── COCO-PanopticSegmentation │ │ │ │ │ │ └── panoptic_deeplab_R_52_os16_mg124_poly_200k_bs64_crop_640_640_coco_dsconv.yaml │ │ │ │ │ └── Cityscapes-PanopticSegmentation │ │ │ │ │ │ ├── Base-PanopticDeepLab-OS16.yaml │ │ │ │ │ │ ├── panoptic_deeplab_R_52_os16_mg124_poly_90k_bs32_crop_512_1024.yaml │ │ │ │ │ │ └── panoptic_deeplab_R_52_os16_mg124_poly_90k_bs32_crop_512_1024_dsconv.yaml │ │ │ │ ├── panoptic_deeplab │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ ├── panoptic_seg.py │ │ │ │ │ ├── post_processing.py │ │ │ │ │ └── target_generator.py │ │ │ │ └── train_net.py │ │ │ ├── PointRend │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── InstanceSegmentation │ │ │ │ │ │ ├── Base-Implicit-PointRend.yaml │ │ │ │ │ │ ├── Base-PointRend-RCNN-FPN.yaml │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_1x_coco.yaml │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_3x_coco.yaml │ │ │ │ │ │ ├── pointrend_rcnn_R_101_FPN_3x_coco.yaml │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_1x_cityscapes.yaml │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_1x_coco.yaml │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_3x_coco.yaml │ │ │ │ │ │ └── pointrend_rcnn_X_101_32x8d_FPN_3x_coco.yaml │ │ │ │ │ └── SemanticSegmentation │ │ │ │ │ │ ├── Base-PointRend-Semantic-FPN.yaml │ │ │ │ │ │ └── pointrend_semantic_R_101_FPN_1x_cityscapes.yaml │ │ │ │ ├── point_rend │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── color_augmentation.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── mask_head.py │ │ │ │ │ ├── point_features.py │ │ │ │ │ ├── point_head.py │ │ │ │ │ ├── roi_heads.py │ │ │ │ │ └── semantic_seg.py │ │ │ │ └── train_net.py │ │ │ ├── PointSup │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_3x_point_sup_point_aug_coco.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_point_sup_coco.yaml │ │ │ │ │ └── mask_rcnn_R_50_FPN_3x_point_sup_point_aug_coco.yaml │ │ │ │ ├── point_sup │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ ├── detection_utils.py │ │ │ │ │ ├── mask_head.py │ │ │ │ │ ├── point_utils.py │ │ │ │ │ └── register_point_annotations.py │ │ │ │ ├── tools │ │ │ │ │ └── prepare_coco_point_annotations_without_masks.py │ │ │ │ └── train_net.py │ │ │ ├── README.md │ │ │ ├── Rethinking-BatchNorm │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── mask_rcnn_BNhead.py │ │ │ │ │ ├── mask_rcnn_BNhead_batch_stats.py │ │ │ │ │ ├── mask_rcnn_BNhead_shuffle.py │ │ │ │ │ ├── mask_rcnn_SyncBNhead.py │ │ │ │ │ ├── retinanet_SyncBNhead.py │ │ │ │ │ └── retinanet_SyncBNhead_SharedTraining.py │ │ │ │ └── retinanet-eval-domain-specific.py │ │ │ ├── TensorMask │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── Base-TensorMask.yaml │ │ │ │ │ ├── tensormask_R_50_FPN_1x.yaml │ │ │ │ │ └── tensormask_R_50_FPN_6x.yaml │ │ │ │ ├── setup.py │ │ │ │ ├── tensormask │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── arch.py │ │ │ │ │ ├── config.py │ │ │ │ │ └── layers │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── csrc │ │ │ │ │ │ ├── SwapAlign2Nat │ │ │ │ │ │ │ ├── SwapAlign2Nat.h │ │ │ │ │ │ │ └── SwapAlign2Nat_cuda.cu │ │ │ │ │ │ └── vision.cpp │ │ │ │ │ │ └── swap_align2nat.py │ │ │ │ ├── tests │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── test_swap_align2nat.py │ │ │ │ └── train_net.py │ │ │ ├── TridentNet │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── Base-TridentNet-Fast-C4.yaml │ │ │ │ │ ├── tridentnet_fast_R_101_C4_3x.yaml │ │ │ │ │ ├── tridentnet_fast_R_50_C4_1x.yaml │ │ │ │ │ └── tridentnet_fast_R_50_C4_3x.yaml │ │ │ │ ├── train_net.py │ │ │ │ └── tridentnet │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── trident_backbone.py │ │ │ │ │ ├── trident_conv.py │ │ │ │ │ ├── trident_rcnn.py │ │ │ │ │ └── trident_rpn.py │ │ │ └── ViTDet │ │ │ │ ├── README.md │ │ │ │ └── configs │ │ │ │ ├── COCO │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_100ep.py │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_36ep.py │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_swin_b_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_swin_l_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_vimdet_b_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vimdet_s_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vimdet_t_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vimdet_t_100ep_adj1.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_b_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_h_75ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_l_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_s_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_t_100ep.py │ │ │ │ ├── lite_cascade_mask_rcnn_vimdet_t_100ep.py │ │ │ │ ├── mask_rcnn_vimdet_b_100ep.py │ │ │ │ ├── mask_rcnn_vitdet_b_100ep.py │ │ │ │ ├── mask_rcnn_vitdet_h_75ep.py │ │ │ │ ├── mask_rcnn_vitdet_l_100ep.py │ │ │ │ └── mask_rcnn_vitdet_t_100ep.py │ │ │ │ ├── LVIS │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_100ep.py │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_swin_b_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_swin_l_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_b_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_h_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_l_100ep.py │ │ │ │ ├── mask_rcnn_vitdet_b_100ep.py │ │ │ │ ├── mask_rcnn_vitdet_h_100ep.py │ │ │ │ └── mask_rcnn_vitdet_l_100ep.py │ │ │ │ └── common │ │ │ │ └── coco_loader_lsj.py │ │ ├── scripts │ │ │ ├── eval_vim_tiny_vimdet.sh │ │ │ └── ft_vim_tiny_vimdet.sh │ │ ├── setup.cfg │ │ ├── setup.py │ │ ├── tests │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── config │ │ │ │ ├── dir1 │ │ │ │ │ ├── bad_import.py │ │ │ │ │ ├── bad_import2.py │ │ │ │ │ ├── dir1_a.py │ │ │ │ │ ├── dir1_b.py │ │ │ │ │ └── load_rel.py │ │ │ │ ├── root_cfg.py │ │ │ │ ├── test_instantiate_config.py │ │ │ │ ├── test_lazy_config.py │ │ │ │ └── test_yacs_config.py │ │ │ ├── data │ │ │ │ ├── __init__.py │ │ │ │ ├── test_coco.py │ │ │ │ ├── test_coco_evaluation.py │ │ │ │ ├── test_dataset.py │ │ │ │ ├── test_detection_utils.py │ │ │ │ ├── test_rotation_transform.py │ │ │ │ ├── test_sampler.py │ │ │ │ └── test_transforms.py │ │ │ ├── export │ │ │ │ └── test_c10.py │ │ │ ├── layers │ │ │ │ ├── __init__.py │ │ │ │ ├── test_blocks.py │ │ │ │ ├── test_deformable.py │ │ │ │ ├── test_losses.py │ │ │ │ ├── test_mask_ops.py │ │ │ │ ├── test_nms.py │ │ │ │ ├── test_nms_rotated.py │ │ │ │ ├── test_roi_align.py │ │ │ │ └── test_roi_align_rotated.py │ │ │ ├── modeling │ │ │ │ ├── __init__.py │ │ │ │ ├── test_anchor_generator.py │ │ │ │ ├── test_backbone.py │ │ │ │ ├── test_box2box_transform.py │ │ │ │ ├── test_fast_rcnn.py │ │ │ │ ├── test_matcher.py │ │ │ │ ├── test_mmdet.py │ │ │ │ ├── test_model_e2e.py │ │ │ │ ├── test_roi_heads.py │ │ │ │ ├── test_roi_pooler.py │ │ │ │ └── test_rpn.py │ │ │ ├── structures │ │ │ │ ├── __init__.py │ │ │ │ ├── test_boxes.py │ │ │ │ ├── test_imagelist.py │ │ │ │ ├── test_instances.py │ │ │ │ ├── test_keypoints.py │ │ │ │ ├── test_masks.py │ │ │ │ └── test_rotated_boxes.py │ │ │ ├── test_checkpoint.py │ │ │ ├── test_engine.py │ │ │ ├── test_events.py │ │ │ ├── test_export_caffe2.py │ │ │ ├── test_export_onnx.py │ │ │ ├── test_export_torchscript.py │ │ │ ├── test_model_analysis.py │ │ │ ├── test_model_zoo.py │ │ │ ├── test_packaging.py │ │ │ ├── test_registry.py │ │ │ ├── test_scheduler.py │ │ │ ├── test_solver.py │ │ │ ├── test_visualizer.py │ │ │ ├── tracking │ │ │ │ ├── __init__.py │ │ │ │ ├── test_bbox_iou_tracker.py │ │ │ │ ├── test_hungarian_tracker.py │ │ │ │ ├── test_iou_weighted_hungarian_bbox_iou_tracker.py │ │ │ │ └── test_vanilla_hungarian_bbox_iou_tracker.py │ │ │ └── utils │ │ │ │ └── test_tensorboardx.py │ │ └── tools │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── analyze_model.py │ │ │ ├── benchmark.py │ │ │ ├── convert-torchvision-to-d2.py │ │ │ ├── deploy │ │ │ ├── CMakeLists.txt │ │ │ ├── README.md │ │ │ ├── export_model.py │ │ │ └── torchscript_mask_rcnn.cpp │ │ │ ├── lazyconfig_train_net.py │ │ │ ├── lightning_train_net.py │ │ │ ├── plain_train_net.py │ │ │ ├── train_net.py │ │ │ ├── visualize_data.py │ │ │ └── visualize_json_results.py │ │ ├── mamba-1p1p1 │ │ ├── .github │ │ │ └── workflows │ │ │ │ └── publish.yaml │ │ ├── .gitignore │ │ ├── .gitmodules │ │ ├── AUTHORS │ │ ├── LICENSE │ │ ├── README.md │ │ ├── assets │ │ │ └── selection.png │ │ ├── benchmarks │ │ │ └── benchmark_generation_mamba_simple.py │ │ ├── csrc │ │ │ └── selective_scan │ │ │ │ ├── reverse_scan.cuh │ │ │ │ ├── selective_scan.cpp │ │ │ │ ├── selective_scan.h │ │ │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ │ │ ├── selective_scan_bwd_bf16_real.cu │ │ │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ │ │ ├── selective_scan_bwd_fp16_real.cu │ │ │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ │ │ ├── selective_scan_bwd_fp32_real.cu │ │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ │ ├── selective_scan_common.h │ │ │ │ ├── selective_scan_fwd_bf16.cu │ │ │ │ ├── selective_scan_fwd_fp16.cu │ │ │ │ ├── selective_scan_fwd_fp32.cu │ │ │ │ ├── selective_scan_fwd_kernel.cuh │ │ │ │ ├── static_switch.h │ │ │ │ └── uninitialized_copy.cuh │ │ ├── evals │ │ │ └── lm_harness_eval.py │ │ ├── mamba_ssm │ │ │ ├── __init__.py │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ ├── config_mamba.py │ │ │ │ └── mixer_seq_simple.py │ │ │ ├── modules │ │ │ │ ├── __init__.py │ │ │ │ └── mamba_simple.py │ │ │ ├── ops │ │ │ │ ├── __init__.py │ │ │ │ ├── selective_scan_interface.py │ │ │ │ └── triton │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── layernorm.py │ │ │ │ │ └── selective_state_update.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── generation.py │ │ │ │ └── hf.py │ │ ├── setup.py │ │ └── tests │ │ │ └── ops │ │ │ ├── test_selective_scan.py │ │ │ └── triton │ │ │ └── test_selective_state_update.py │ │ ├── seg │ │ ├── README.md │ │ ├── backbone │ │ │ ├── __init__.py │ │ │ └── vim.py │ │ ├── configs │ │ │ ├── _base_ │ │ │ │ ├── datasets │ │ │ │ │ ├── ade20k.py │ │ │ │ │ ├── ade20k_1024x1024.py │ │ │ │ │ ├── ade20k_640x640.py │ │ │ │ │ └── ade20k_ms_eval.py │ │ │ │ ├── default_runtime.py │ │ │ │ ├── models │ │ │ │ │ ├── plainseg_vim.py │ │ │ │ │ ├── upernet_beit.py │ │ │ │ │ └── upernet_vim.py │ │ │ │ └── schedules │ │ │ │ │ ├── schedule_60k.py │ │ │ │ │ └── schedule_80k.py │ │ │ └── vim │ │ │ │ └── upernet │ │ │ │ ├── upernet_vim_small_24_512_slide_60k.py │ │ │ │ └── upernet_vim_tiny_24_512_slide_60k.py │ │ ├── mmcv_custom │ │ │ ├── __init__.py │ │ │ ├── apex_runner │ │ │ │ ├── __init__.py │ │ │ │ ├── apex_iter_based_runner.py │ │ │ │ ├── checkpoint.py │ │ │ │ └── optimizer.py │ │ │ ├── checkpoint.py │ │ │ ├── layer_decay_optimizer_constructor.py │ │ │ ├── resize_transform.py │ │ │ └── train_api.py │ │ ├── scripts │ │ │ ├── eval_vim_small_upernet.sh │ │ │ ├── eval_vim_tiny_upernet.sh │ │ │ ├── ft_vim_small_upernet.sh │ │ │ └── ft_vim_tiny_upernet.sh │ │ ├── seg-requirements.txt │ │ ├── test.py │ │ ├── tools │ │ │ └── analysis_tools │ │ │ │ ├── benchmark.py │ │ │ │ ├── benchmark_scripts.sh │ │ │ │ ├── deit-t-upernet.txt │ │ │ │ ├── plot_curve.ipynb │ │ │ │ └── vim-t-upernet.txt │ │ └── train.py │ │ └── vim │ │ ├── .gitignore │ │ ├── =0.9.0 │ │ ├── =1.8.0 │ │ ├── LICENSE │ │ ├── augment.py │ │ ├── calc_flops.py │ │ ├── datasets.py │ │ ├── engine.py │ │ ├── hubconf.py │ │ ├── losses.py │ │ ├── main.py │ │ ├── models │ │ ├── models_mamba.py │ │ ├── videomamba.py │ │ └── vim.py │ │ ├── rope.py │ │ ├── run_with_submitit.py │ │ ├── samplers.py │ │ ├── scripts │ │ └── ft-vim-t.sh │ │ ├── utils.py │ │ ├── vim_requirements.txt │ │ └── visualization.ipynb ├── Pruning_Strategy │ ├── .gitignore │ ├── Random │ │ ├── .gitignore │ │ ├── =1.1.0 │ │ ├── LICENSE.txt │ │ ├── README.md │ │ ├── assets │ │ │ ├── vim_pipeline_v1.9.png │ │ │ └── vim_teaser_v1.7.png │ │ ├── det │ │ │ ├── .circleci │ │ │ │ ├── config.yml │ │ │ │ └── import-tests.sh │ │ │ ├── .clang-format │ │ │ ├── .flake8 │ │ │ ├── .gitignore │ │ │ ├── GETTING_STARTED.md │ │ │ ├── INSTALL.md │ │ │ ├── LICENSE │ │ │ ├── MODEL_ZOO.md │ │ │ ├── README.md │ │ │ ├── configs │ │ │ │ ├── Base-RCNN-C4.yaml │ │ │ │ ├── Base-RCNN-DilatedC5.yaml │ │ │ │ ├── Base-RCNN-FPN.yaml │ │ │ │ ├── Base-RetinaNet.yaml │ │ │ │ ├── COCO-Detection │ │ │ │ │ ├── fast_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── faster_rcnn_R_101_C4_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_101_DC5_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_101_FPN_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_C4_1x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_C4_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_DC5_1x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_DC5_3x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── faster_rcnn_R_50_FPN_3x.yaml │ │ │ │ │ ├── faster_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ │ ├── fcos_R_50_FPN_1x.py │ │ │ │ │ ├── retinanet_R_101_FPN_3x.yaml │ │ │ │ │ ├── retinanet_R_50_FPN_1x.py │ │ │ │ │ ├── retinanet_R_50_FPN_1x.yaml │ │ │ │ │ ├── retinanet_R_50_FPN_3x.yaml │ │ │ │ │ ├── rpn_R_50_C4_1x.yaml │ │ │ │ │ └── rpn_R_50_FPN_1x.yaml │ │ │ │ ├── COCO-InstanceSegmentation │ │ │ │ │ ├── mask_rcnn_R_101_C4_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_101_DC5_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_101_FPN_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_1x.py │ │ │ │ │ ├── mask_rcnn_R_50_C4_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_DC5_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_DC5_3x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_giou.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x.yaml │ │ │ │ │ ├── mask_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_fpn_1x.py │ │ │ │ │ └── mask_rcnn_regnety_4gf_dds_fpn_1x.py │ │ │ │ ├── COCO-Keypoints │ │ │ │ │ ├── Base-Keypoint-RCNN-FPN.yaml │ │ │ │ │ ├── keypoint_rcnn_R_101_FPN_3x.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.py │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_3x.yaml │ │ │ │ │ └── keypoint_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ ├── COCO-PanopticSegmentation │ │ │ │ │ ├── Base-Panoptic-FPN.yaml │ │ │ │ │ ├── panoptic_fpn_R_101_3x.yaml │ │ │ │ │ ├── panoptic_fpn_R_50_1x.py │ │ │ │ │ ├── panoptic_fpn_R_50_1x.yaml │ │ │ │ │ └── panoptic_fpn_R_50_3x.yaml │ │ │ │ ├── Cityscapes │ │ │ │ │ └── mask_rcnn_R_50_FPN.yaml │ │ │ │ ├── Detectron1-Comparisons │ │ │ │ │ ├── README.md │ │ │ │ │ ├── faster_rcnn_R_50_FPN_noaug_1x.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ └── mask_rcnn_R_50_FPN_noaug_1x.yaml │ │ │ │ ├── LVISv0.5-InstanceSegmentation │ │ │ │ │ ├── mask_rcnn_R_101_FPN_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ └── mask_rcnn_X_101_32x8d_FPN_1x.yaml │ │ │ │ ├── LVISv1-InstanceSegmentation │ │ │ │ │ ├── mask_rcnn_R_101_FPN_1x.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ └── mask_rcnn_X_101_32x8d_FPN_1x.yaml │ │ │ │ ├── Misc │ │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_3x.yaml │ │ │ │ │ ├── cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_cls_agnostic.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_dconv_c3-c5.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_dconv_c3-c5.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_gn.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_syncbn.yaml │ │ │ │ │ ├── mmdet_mask_rcnn_R_50_FPN_1x.py │ │ │ │ │ ├── panoptic_fpn_R_101_dconv_cascade_gn_3x.yaml │ │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_3x_gn.yaml │ │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_9x_gn.yaml │ │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_9x_syncbn.yaml │ │ │ │ │ ├── semantic_R_50_FPN_1x.yaml │ │ │ │ │ └── torchvision_imagenet_R_50.py │ │ │ │ ├── PascalVOC-Detection │ │ │ │ │ ├── faster_rcnn_R_50_C4.yaml │ │ │ │ │ └── faster_rcnn_R_50_FPN.yaml │ │ │ │ ├── common │ │ │ │ │ ├── README.md │ │ │ │ │ ├── coco_schedule.py │ │ │ │ │ ├── data │ │ │ │ │ │ ├── coco.py │ │ │ │ │ │ ├── coco_keypoint.py │ │ │ │ │ │ ├── coco_panoptic_separated.py │ │ │ │ │ │ └── constants.py │ │ │ │ │ ├── models │ │ │ │ │ │ ├── cascade_rcnn.py │ │ │ │ │ │ ├── fcos.py │ │ │ │ │ │ ├── keypoint_rcnn_fpn.py │ │ │ │ │ │ ├── mask_rcnn_c4.py │ │ │ │ │ │ ├── mask_rcnn_fpn.py │ │ │ │ │ │ ├── mask_rcnn_vimdet.py │ │ │ │ │ │ ├── mask_rcnn_vitdet.py │ │ │ │ │ │ ├── panoptic_fpn.py │ │ │ │ │ │ └── retinanet.py │ │ │ │ │ ├── optim.py │ │ │ │ │ └── train.py │ │ │ │ ├── new_baselines │ │ │ │ │ ├── mask_rcnn_R_101_FPN_100ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_101_FPN_200ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_101_FPN_400ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_100ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_200ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_400ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_R_50_FPN_50ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_100ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_200ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_400ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnety_4gf_dds_FPN_100ep_LSJ.py │ │ │ │ │ ├── mask_rcnn_regnety_4gf_dds_FPN_200ep_LSJ.py │ │ │ │ │ └── mask_rcnn_regnety_4gf_dds_FPN_400ep_LSJ.py │ │ │ │ └── quick_schedules │ │ │ │ │ ├── README.md │ │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── fast_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── fast_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_normalized_training_acc_test.yaml │ │ │ │ │ ├── keypoint_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_GCV_instant_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_inference_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_instant_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_C4_training_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_DC5_inference_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_pred_boxes_training_acc_test.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ │ ├── panoptic_fpn_R_50_inference_acc_test.yaml │ │ │ │ │ ├── panoptic_fpn_R_50_instant_test.yaml │ │ │ │ │ ├── panoptic_fpn_R_50_training_acc_test.yaml │ │ │ │ │ ├── retinanet_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── retinanet_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── rpn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── rpn_R_50_FPN_instant_test.yaml │ │ │ │ │ ├── semantic_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ ├── semantic_R_50_FPN_instant_test.yaml │ │ │ │ │ └── semantic_R_50_FPN_training_acc_test.yaml │ │ │ ├── datasets │ │ │ │ ├── README.md │ │ │ │ ├── prepare_ade20k_sem_seg.py │ │ │ │ ├── prepare_cocofied_lvis.py │ │ │ │ ├── prepare_for_tests.sh │ │ │ │ └── prepare_panoptic_fpn.py │ │ │ ├── demo │ │ │ │ ├── README.md │ │ │ │ ├── demo.py │ │ │ │ └── predictor.py │ │ │ ├── det-requirements.txt │ │ │ ├── detectron2 │ │ │ │ ├── __init__.py │ │ │ │ ├── checkpoint │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── c2_model_loading.py │ │ │ │ │ ├── catalog.py │ │ │ │ │ └── detection_checkpoint.py │ │ │ │ ├── config │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── compat.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── defaults.py │ │ │ │ │ ├── instantiate.py │ │ │ │ │ └── lazy.py │ │ │ │ ├── data │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── benchmark.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── catalog.py │ │ │ │ │ ├── common.py │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ ├── datasets │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ ├── builtin_meta.py │ │ │ │ │ │ ├── cityscapes.py │ │ │ │ │ │ ├── cityscapes_panoptic.py │ │ │ │ │ │ ├── coco.py │ │ │ │ │ │ ├── coco_panoptic.py │ │ │ │ │ │ ├── lvis.py │ │ │ │ │ │ ├── lvis_v0_5_categories.py │ │ │ │ │ │ ├── lvis_v1_categories.py │ │ │ │ │ │ ├── lvis_v1_category_image_count.py │ │ │ │ │ │ ├── pascal_voc.py │ │ │ │ │ │ └── register_coco.py │ │ │ │ │ ├── detection_utils.py │ │ │ │ │ ├── samplers │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── distributed_sampler.py │ │ │ │ │ │ └── grouped_batch_sampler.py │ │ │ │ │ └── transforms │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── augmentation.py │ │ │ │ │ │ ├── augmentation_impl.py │ │ │ │ │ │ └── transform.py │ │ │ │ ├── engine │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── defaults.py │ │ │ │ │ ├── hooks.py │ │ │ │ │ ├── launch.py │ │ │ │ │ └── train_loop.py │ │ │ │ ├── evaluation │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── cityscapes_evaluation.py │ │ │ │ │ ├── coco_evaluation.py │ │ │ │ │ ├── evaluator.py │ │ │ │ │ ├── fast_eval_api.py │ │ │ │ │ ├── lvis_evaluation.py │ │ │ │ │ ├── panoptic_evaluation.py │ │ │ │ │ ├── pascal_voc_evaluation.py │ │ │ │ │ ├── rotated_coco_evaluation.py │ │ │ │ │ ├── sem_seg_evaluation.py │ │ │ │ │ └── testing.py │ │ │ │ ├── export │ │ │ │ │ ├── README.md │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── api.py │ │ │ │ │ ├── c10.py │ │ │ │ │ ├── caffe2_export.py │ │ │ │ │ ├── caffe2_inference.py │ │ │ │ │ ├── caffe2_modeling.py │ │ │ │ │ ├── caffe2_patch.py │ │ │ │ │ ├── flatten.py │ │ │ │ │ ├── shared.py │ │ │ │ │ ├── torchscript.py │ │ │ │ │ └── torchscript_patch.py │ │ │ │ ├── layers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── aspp.py │ │ │ │ │ ├── batch_norm.py │ │ │ │ │ ├── blocks.py │ │ │ │ │ ├── csrc │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── ROIAlignRotated │ │ │ │ │ │ │ ├── ROIAlignRotated.h │ │ │ │ │ │ │ ├── ROIAlignRotated_cpu.cpp │ │ │ │ │ │ │ └── ROIAlignRotated_cuda.cu │ │ │ │ │ │ ├── box_iou_rotated │ │ │ │ │ │ │ ├── box_iou_rotated.h │ │ │ │ │ │ │ ├── box_iou_rotated_cpu.cpp │ │ │ │ │ │ │ ├── box_iou_rotated_cuda.cu │ │ │ │ │ │ │ └── box_iou_rotated_utils.h │ │ │ │ │ │ ├── cocoeval │ │ │ │ │ │ │ ├── cocoeval.cpp │ │ │ │ │ │ │ └── cocoeval.h │ │ │ │ │ │ ├── cuda_version.cu │ │ │ │ │ │ ├── deformable │ │ │ │ │ │ │ ├── deform_conv.h │ │ │ │ │ │ │ ├── deform_conv_cuda.cu │ │ │ │ │ │ │ └── deform_conv_cuda_kernel.cu │ │ │ │ │ │ ├── nms_rotated │ │ │ │ │ │ │ ├── nms_rotated.h │ │ │ │ │ │ │ ├── nms_rotated_cpu.cpp │ │ │ │ │ │ │ └── nms_rotated_cuda.cu │ │ │ │ │ │ └── vision.cpp │ │ │ │ │ ├── deform_conv.py │ │ │ │ │ ├── losses.py │ │ │ │ │ ├── mask_ops.py │ │ │ │ │ ├── nms.py │ │ │ │ │ ├── roi_align.py │ │ │ │ │ ├── roi_align_rotated.py │ │ │ │ │ ├── rotated_boxes.py │ │ │ │ │ ├── shape_spec.py │ │ │ │ │ └── wrappers.py │ │ │ │ ├── model_zoo │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── model_zoo.py │ │ │ │ ├── modeling │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── anchor_generator.py │ │ │ │ │ ├── backbone │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── backbone.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── fpn.py │ │ │ │ │ │ ├── mvit.py │ │ │ │ │ │ ├── regnet.py │ │ │ │ │ │ ├── resnet.py │ │ │ │ │ │ ├── swin.py │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ ├── vim.py │ │ │ │ │ │ └── vit.py │ │ │ │ │ ├── box_regression.py │ │ │ │ │ ├── matcher.py │ │ │ │ │ ├── meta_arch │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── dense_detector.py │ │ │ │ │ │ ├── fcos.py │ │ │ │ │ │ ├── panoptic_fpn.py │ │ │ │ │ │ ├── rcnn.py │ │ │ │ │ │ ├── retinanet.py │ │ │ │ │ │ └── semantic_seg.py │ │ │ │ │ ├── mmdet_wrapper.py │ │ │ │ │ ├── poolers.py │ │ │ │ │ ├── postprocessing.py │ │ │ │ │ ├── proposal_generator │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── proposal_utils.py │ │ │ │ │ │ ├── rpn.py │ │ │ │ │ │ └── rrpn.py │ │ │ │ │ ├── roi_heads │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── box_head.py │ │ │ │ │ │ ├── cascade_rcnn.py │ │ │ │ │ │ ├── fast_rcnn.py │ │ │ │ │ │ ├── keypoint_head.py │ │ │ │ │ │ ├── mask_head.py │ │ │ │ │ │ ├── roi_heads.py │ │ │ │ │ │ └── rotated_fast_rcnn.py │ │ │ │ │ ├── sampling.py │ │ │ │ │ └── test_time_augmentation.py │ │ │ │ ├── projects │ │ │ │ │ ├── README.md │ │ │ │ │ └── __init__.py │ │ │ │ ├── solver │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build.py │ │ │ │ │ └── lr_scheduler.py │ │ │ │ ├── structures │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── boxes.py │ │ │ │ │ ├── image_list.py │ │ │ │ │ ├── instances.py │ │ │ │ │ ├── keypoints.py │ │ │ │ │ ├── masks.py │ │ │ │ │ └── rotated_boxes.py │ │ │ │ ├── tracking │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_tracker.py │ │ │ │ │ ├── bbox_iou_tracker.py │ │ │ │ │ ├── hungarian_tracker.py │ │ │ │ │ ├── iou_weighted_hungarian_bbox_iou_tracker.py │ │ │ │ │ ├── utils.py │ │ │ │ │ └── vanilla_hungarian_bbox_iou_tracker.py │ │ │ │ └── utils │ │ │ │ │ ├── README.md │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── analysis.py │ │ │ │ │ ├── collect_env.py │ │ │ │ │ ├── colormap.py │ │ │ │ │ ├── comm.py │ │ │ │ │ ├── develop.py │ │ │ │ │ ├── env.py │ │ │ │ │ ├── events.py │ │ │ │ │ ├── file_io.py │ │ │ │ │ ├── logger.py │ │ │ │ │ ├── memory.py │ │ │ │ │ ├── registry.py │ │ │ │ │ ├── serialize.py │ │ │ │ │ ├── testing.py │ │ │ │ │ ├── tracing.py │ │ │ │ │ ├── video_visualizer.py │ │ │ │ │ └── visualizer.py │ │ │ ├── dev │ │ │ │ ├── README.md │ │ │ │ ├── linter.sh │ │ │ │ ├── packaging │ │ │ │ │ ├── README.md │ │ │ │ │ ├── build_all_wheels.sh │ │ │ │ │ ├── build_wheel.sh │ │ │ │ │ ├── gen_install_table.py │ │ │ │ │ ├── gen_wheel_index.sh │ │ │ │ │ └── pkg_helpers.bash │ │ │ │ ├── parse_results.sh │ │ │ │ ├── run_inference_tests.sh │ │ │ │ └── run_instant_tests.sh │ │ │ ├── docker │ │ │ │ ├── Dockerfile │ │ │ │ ├── README.md │ │ │ │ ├── deploy.Dockerfile │ │ │ │ └── docker-compose.yml │ │ │ ├── docs │ │ │ │ ├── .gitignore │ │ │ │ ├── Makefile │ │ │ │ ├── README.md │ │ │ │ ├── _static │ │ │ │ │ └── css │ │ │ │ │ │ └── custom.css │ │ │ │ ├── conf.py │ │ │ │ ├── index.rst │ │ │ │ ├── modules │ │ │ │ │ ├── checkpoint.rst │ │ │ │ │ ├── config.rst │ │ │ │ │ ├── data.rst │ │ │ │ │ ├── data_transforms.rst │ │ │ │ │ ├── engine.rst │ │ │ │ │ ├── evaluation.rst │ │ │ │ │ ├── export.rst │ │ │ │ │ ├── fvcore.rst │ │ │ │ │ ├── index.rst │ │ │ │ │ ├── layers.rst │ │ │ │ │ ├── model_zoo.rst │ │ │ │ │ ├── modeling.rst │ │ │ │ │ ├── solver.rst │ │ │ │ │ ├── structures.rst │ │ │ │ │ └── utils.rst │ │ │ │ ├── notes │ │ │ │ │ ├── benchmarks.md │ │ │ │ │ ├── changelog.md │ │ │ │ │ ├── compatibility.md │ │ │ │ │ ├── contributing.md │ │ │ │ │ └── index.rst │ │ │ │ ├── requirements.txt │ │ │ │ └── tutorials │ │ │ │ │ ├── README.md │ │ │ │ │ ├── augmentation.md │ │ │ │ │ ├── builtin_datasets.md │ │ │ │ │ ├── configs.md │ │ │ │ │ ├── data_loading.md │ │ │ │ │ ├── datasets.md │ │ │ │ │ ├── deployment.md │ │ │ │ │ ├── evaluation.md │ │ │ │ │ ├── extend.md │ │ │ │ │ ├── getting_started.md │ │ │ │ │ ├── index.rst │ │ │ │ │ ├── install.md │ │ │ │ │ ├── lazyconfigs.md │ │ │ │ │ ├── models.md │ │ │ │ │ ├── training.md │ │ │ │ │ └── write-models.md │ │ │ ├── projects │ │ │ │ ├── DeepLab │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ └── Cityscapes-SemanticSegmentation │ │ │ │ │ │ │ ├── Base-DeepLabV3-OS16-Semantic.yaml │ │ │ │ │ │ │ ├── deeplab_v3_R_103_os16_mg124_poly_90k_bs16.yaml │ │ │ │ │ │ │ └── deeplab_v3_plus_R_103_os16_mg124_poly_90k_bs16.yaml │ │ │ │ │ ├── deeplab │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build_solver.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── loss.py │ │ │ │ │ │ ├── lr_scheduler.py │ │ │ │ │ │ ├── resnet.py │ │ │ │ │ │ └── semantic_seg.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── DensePose │ │ │ │ │ ├── README.md │ │ │ │ │ ├── apply_net.py │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN.yaml │ │ │ │ │ │ ├── HRNet │ │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w32_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w40_s1x.yaml │ │ │ │ │ │ │ └── densepose_rcnn_HRFPN_HRNet_w48_s1x.yaml │ │ │ │ │ │ ├── cse │ │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN-Human.yaml │ │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_soft_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_soft_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_soft_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_4k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_i2m_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_m2m_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_16k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_4k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_maskonly_24k.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_chimps_finetune_4k.yaml │ │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_soft_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC1M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC1_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC2M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC2_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC1M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC1_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC2M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC2_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x_legacy.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC1M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC1_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC2M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC2_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2M_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x_legacy.yaml │ │ │ │ │ │ ├── evolution │ │ │ │ │ │ │ ├── Base-RCNN-FPN-Atop10P_CA.yaml │ │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA.yaml │ │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_coarsesegm.yaml │ │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_finesegm.yaml │ │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_uniform.yaml │ │ │ │ │ │ │ └── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_uv.yaml │ │ │ │ │ │ └── quick_schedules │ │ │ │ │ │ │ ├── cse │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_instant_test.yaml │ │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_soft_animals_finetune_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w32_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_TTA_inference_acc_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2_instant_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ │ ├── densepose │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── converters │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── base.py │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ ├── chart_output_hflip.py │ │ │ │ │ │ │ ├── chart_output_to_chart_result.py │ │ │ │ │ │ │ ├── hflip.py │ │ │ │ │ │ │ ├── segm_to_mask.py │ │ │ │ │ │ │ ├── to_chart_result.py │ │ │ │ │ │ │ └── to_mask.py │ │ │ │ │ │ ├── data │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ │ ├── combined_loader.py │ │ │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ │ │ ├── datasets │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ │ ├── chimpnsee.py │ │ │ │ │ │ │ │ ├── coco.py │ │ │ │ │ │ │ │ ├── dataset_type.py │ │ │ │ │ │ │ │ └── lvis.py │ │ │ │ │ │ │ ├── image_list_dataset.py │ │ │ │ │ │ │ ├── inference_based_loader.py │ │ │ │ │ │ │ ├── meshes │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ │ └── catalog.py │ │ │ │ │ │ │ ├── samplers │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── densepose_base.py │ │ │ │ │ │ │ │ ├── densepose_confidence_based.py │ │ │ │ │ │ │ │ ├── densepose_cse_base.py │ │ │ │ │ │ │ │ ├── densepose_cse_confidence_based.py │ │ │ │ │ │ │ │ ├── densepose_cse_uniform.py │ │ │ │ │ │ │ │ ├── densepose_uniform.py │ │ │ │ │ │ │ │ ├── mask_from_densepose.py │ │ │ │ │ │ │ │ └── prediction_to_gt.py │ │ │ │ │ │ │ ├── transform │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ └── image.py │ │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ │ └── video │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── frame_selector.py │ │ │ │ │ │ │ │ └── video_keyframe_dataset.py │ │ │ │ │ │ ├── engine │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ └── trainer.py │ │ │ │ │ │ ├── evaluation │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── d2_evaluator_adapter.py │ │ │ │ │ │ │ ├── densepose_coco_evaluation.py │ │ │ │ │ │ │ ├── evaluator.py │ │ │ │ │ │ │ ├── mesh_alignment_evaluator.py │ │ │ │ │ │ │ └── tensor_storage.py │ │ │ │ │ │ ├── modeling │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ │ ├── confidence.py │ │ │ │ │ │ │ ├── cse │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── embedder.py │ │ │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ │ │ ├── vertex_direct_embedder.py │ │ │ │ │ │ │ │ └── vertex_feature_embedder.py │ │ │ │ │ │ │ ├── densepose_checkpoint.py │ │ │ │ │ │ │ ├── filter.py │ │ │ │ │ │ │ ├── hrfpn.py │ │ │ │ │ │ │ ├── hrnet.py │ │ │ │ │ │ │ ├── inference.py │ │ │ │ │ │ │ ├── losses │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ │ ├── chart_with_confidences.py │ │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ │ ├── cycle_pix2shape.py │ │ │ │ │ │ │ │ ├── cycle_shape2shape.py │ │ │ │ │ │ │ │ ├── embed.py │ │ │ │ │ │ │ │ ├── embed_utils.py │ │ │ │ │ │ │ │ ├── mask.py │ │ │ │ │ │ │ │ ├── mask_or_segm.py │ │ │ │ │ │ │ │ ├── registry.py │ │ │ │ │ │ │ │ ├── segm.py │ │ │ │ │ │ │ │ ├── soft_embed.py │ │ │ │ │ │ │ │ └── utils.py │ │ │ │ │ │ │ ├── predictors │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ │ ├── chart_confidence.py │ │ │ │ │ │ │ │ ├── chart_with_confidence.py │ │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ │ ├── cse_confidence.py │ │ │ │ │ │ │ │ ├── cse_with_confidence.py │ │ │ │ │ │ │ │ └── registry.py │ │ │ │ │ │ │ ├── roi_heads │ │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ │ ├── deeplab.py │ │ │ │ │ │ │ │ ├── registry.py │ │ │ │ │ │ │ │ ├── roi_head.py │ │ │ │ │ │ │ │ └── v1convx.py │ │ │ │ │ │ │ ├── test_time_augmentation.py │ │ │ │ │ │ │ └── utils.py │ │ │ │ │ │ ├── structures │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ ├── chart_confidence.py │ │ │ │ │ │ │ ├── chart_result.py │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ ├── cse_confidence.py │ │ │ │ │ │ │ ├── data_relative.py │ │ │ │ │ │ │ ├── list.py │ │ │ │ │ │ │ ├── mesh.py │ │ │ │ │ │ │ └── transform_data.py │ │ │ │ │ │ ├── utils │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── dbhelper.py │ │ │ │ │ │ │ ├── logger.py │ │ │ │ │ │ │ └── transform.py │ │ │ │ │ │ └── vis │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── base.py │ │ │ │ │ │ │ ├── bounding_box.py │ │ │ │ │ │ │ ├── densepose_data_points.py │ │ │ │ │ │ │ ├── densepose_outputs_iuv.py │ │ │ │ │ │ │ ├── densepose_outputs_vertex.py │ │ │ │ │ │ │ ├── densepose_results.py │ │ │ │ │ │ │ ├── densepose_results_textures.py │ │ │ │ │ │ │ └── extractor.py │ │ │ │ │ ├── dev │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── run_inference_tests.sh │ │ │ │ │ │ └── run_instant_tests.sh │ │ │ │ │ ├── doc │ │ │ │ │ │ ├── BOOTSTRAPPING_PIPELINE.md │ │ │ │ │ │ ├── DENSEPOSE_CSE.md │ │ │ │ │ │ ├── DENSEPOSE_DATASETS.md │ │ │ │ │ │ ├── DENSEPOSE_IUV.md │ │ │ │ │ │ ├── GETTING_STARTED.md │ │ │ │ │ │ ├── RELEASE_2020_04.md │ │ │ │ │ │ ├── RELEASE_2021_03.md │ │ │ │ │ │ ├── RELEASE_2021_06.md │ │ │ │ │ │ ├── TOOL_APPLY_NET.md │ │ │ │ │ │ └── TOOL_QUERY_DB.md │ │ │ │ │ ├── query_db.py │ │ │ │ │ ├── setup.py │ │ │ │ │ ├── tests │ │ │ │ │ │ ├── common.py │ │ │ │ │ │ ├── test_chart_based_annotations_accumulator.py │ │ │ │ │ │ ├── test_combine_data_loader.py │ │ │ │ │ │ ├── test_cse_annotations_accumulator.py │ │ │ │ │ │ ├── test_dataset_loaded_annotations.py │ │ │ │ │ │ ├── test_frame_selector.py │ │ │ │ │ │ ├── test_image_list_dataset.py │ │ │ │ │ │ ├── test_image_resize_transform.py │ │ │ │ │ │ ├── test_model_e2e.py │ │ │ │ │ │ ├── test_setup.py │ │ │ │ │ │ ├── test_structures.py │ │ │ │ │ │ ├── test_tensor_storage.py │ │ │ │ │ │ └── test_video_keyframe_dataset.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── MViTv2 │ │ │ │ │ ├── README.md │ │ │ │ │ └── configs │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_3x.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_3x.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_lsj_3x.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_lsj_50ep.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_s_3x.py │ │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_t_3x.py │ │ │ │ │ │ ├── common │ │ │ │ │ │ ├── coco_loader.py │ │ │ │ │ │ └── coco_loader_lsj.py │ │ │ │ │ │ └── mask_rcnn_mvitv2_t_3x.py │ │ │ │ ├── Panoptic-DeepLab │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── COCO-PanopticSegmentation │ │ │ │ │ │ │ └── panoptic_deeplab_R_52_os16_mg124_poly_200k_bs64_crop_640_640_coco_dsconv.yaml │ │ │ │ │ │ └── Cityscapes-PanopticSegmentation │ │ │ │ │ │ │ ├── Base-PanopticDeepLab-OS16.yaml │ │ │ │ │ │ │ ├── panoptic_deeplab_R_52_os16_mg124_poly_90k_bs32_crop_512_1024.yaml │ │ │ │ │ │ │ └── panoptic_deeplab_R_52_os16_mg124_poly_90k_bs32_crop_512_1024_dsconv.yaml │ │ │ │ │ ├── panoptic_deeplab │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ │ ├── panoptic_seg.py │ │ │ │ │ │ ├── post_processing.py │ │ │ │ │ │ └── target_generator.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── PointRend │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── InstanceSegmentation │ │ │ │ │ │ │ ├── Base-Implicit-PointRend.yaml │ │ │ │ │ │ │ ├── Base-PointRend-RCNN-FPN.yaml │ │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_1x_coco.yaml │ │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_3x_coco.yaml │ │ │ │ │ │ │ ├── pointrend_rcnn_R_101_FPN_3x_coco.yaml │ │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_1x_cityscapes.yaml │ │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_1x_coco.yaml │ │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_3x_coco.yaml │ │ │ │ │ │ │ └── pointrend_rcnn_X_101_32x8d_FPN_3x_coco.yaml │ │ │ │ │ │ └── SemanticSegmentation │ │ │ │ │ │ │ ├── Base-PointRend-Semantic-FPN.yaml │ │ │ │ │ │ │ └── pointrend_semantic_R_101_FPN_1x_cityscapes.yaml │ │ │ │ │ ├── point_rend │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── color_augmentation.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── mask_head.py │ │ │ │ │ │ ├── point_features.py │ │ │ │ │ │ ├── point_head.py │ │ │ │ │ │ ├── roi_heads.py │ │ │ │ │ │ └── semantic_seg.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── PointSup │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_3x_point_sup_point_aug_coco.yaml │ │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_point_sup_coco.yaml │ │ │ │ │ │ └── mask_rcnn_R_50_FPN_3x_point_sup_point_aug_coco.yaml │ │ │ │ │ ├── point_sup │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ │ ├── detection_utils.py │ │ │ │ │ │ ├── mask_head.py │ │ │ │ │ │ ├── point_utils.py │ │ │ │ │ │ └── register_point_annotations.py │ │ │ │ │ ├── tools │ │ │ │ │ │ └── prepare_coco_point_annotations_without_masks.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── README.md │ │ │ │ ├── Rethinking-BatchNorm │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── mask_rcnn_BNhead.py │ │ │ │ │ │ ├── mask_rcnn_BNhead_batch_stats.py │ │ │ │ │ │ ├── mask_rcnn_BNhead_shuffle.py │ │ │ │ │ │ ├── mask_rcnn_SyncBNhead.py │ │ │ │ │ │ ├── retinanet_SyncBNhead.py │ │ │ │ │ │ └── retinanet_SyncBNhead_SharedTraining.py │ │ │ │ │ └── retinanet-eval-domain-specific.py │ │ │ │ ├── TensorMask │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── Base-TensorMask.yaml │ │ │ │ │ │ ├── tensormask_R_50_FPN_1x.yaml │ │ │ │ │ │ └── tensormask_R_50_FPN_6x.yaml │ │ │ │ │ ├── setup.py │ │ │ │ │ ├── tensormask │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── arch.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ └── layers │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── csrc │ │ │ │ │ │ │ ├── SwapAlign2Nat │ │ │ │ │ │ │ │ ├── SwapAlign2Nat.h │ │ │ │ │ │ │ │ └── SwapAlign2Nat_cuda.cu │ │ │ │ │ │ │ └── vision.cpp │ │ │ │ │ │ │ └── swap_align2nat.py │ │ │ │ │ ├── tests │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── test_swap_align2nat.py │ │ │ │ │ └── train_net.py │ │ │ │ ├── TridentNet │ │ │ │ │ ├── README.md │ │ │ │ │ ├── configs │ │ │ │ │ │ ├── Base-TridentNet-Fast-C4.yaml │ │ │ │ │ │ ├── tridentnet_fast_R_101_C4_3x.yaml │ │ │ │ │ │ ├── tridentnet_fast_R_50_C4_1x.yaml │ │ │ │ │ │ └── tridentnet_fast_R_50_C4_3x.yaml │ │ │ │ │ ├── train_net.py │ │ │ │ │ └── tridentnet │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── config.py │ │ │ │ │ │ ├── trident_backbone.py │ │ │ │ │ │ ├── trident_conv.py │ │ │ │ │ │ ├── trident_rcnn.py │ │ │ │ │ │ └── trident_rpn.py │ │ │ │ └── ViTDet │ │ │ │ │ ├── README.md │ │ │ │ │ └── configs │ │ │ │ │ ├── COCO │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_36ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_swin_b_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_swin_l_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vimdet_b_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vimdet_s_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vimdet_t_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vimdet_t_100ep_adj1.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_b_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_h_75ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_l_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_s_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_t_100ep.py │ │ │ │ │ ├── lite_cascade_mask_rcnn_vimdet_t_100ep.py │ │ │ │ │ ├── mask_rcnn_vimdet_b_100ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_b_100ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_h_75ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_l_100ep.py │ │ │ │ │ └── mask_rcnn_vitdet_t_100ep.py │ │ │ │ │ ├── LVIS │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_swin_b_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_swin_l_in21k_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_b_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_h_100ep.py │ │ │ │ │ ├── cascade_mask_rcnn_vitdet_l_100ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_b_100ep.py │ │ │ │ │ ├── mask_rcnn_vitdet_h_100ep.py │ │ │ │ │ └── mask_rcnn_vitdet_l_100ep.py │ │ │ │ │ └── common │ │ │ │ │ └── coco_loader_lsj.py │ │ │ ├── scripts │ │ │ │ ├── eval_vim_tiny_vimdet.sh │ │ │ │ └── ft_vim_tiny_vimdet.sh │ │ │ ├── setup.cfg │ │ │ ├── setup.py │ │ │ ├── tests │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── config │ │ │ │ │ ├── dir1 │ │ │ │ │ │ ├── bad_import.py │ │ │ │ │ │ ├── bad_import2.py │ │ │ │ │ │ ├── dir1_a.py │ │ │ │ │ │ ├── dir1_b.py │ │ │ │ │ │ └── load_rel.py │ │ │ │ │ ├── root_cfg.py │ │ │ │ │ ├── test_instantiate_config.py │ │ │ │ │ ├── test_lazy_config.py │ │ │ │ │ └── test_yacs_config.py │ │ │ │ ├── data │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_coco.py │ │ │ │ │ ├── test_coco_evaluation.py │ │ │ │ │ ├── test_dataset.py │ │ │ │ │ ├── test_detection_utils.py │ │ │ │ │ ├── test_rotation_transform.py │ │ │ │ │ ├── test_sampler.py │ │ │ │ │ └── test_transforms.py │ │ │ │ ├── export │ │ │ │ │ └── test_c10.py │ │ │ │ ├── layers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_blocks.py │ │ │ │ │ ├── test_deformable.py │ │ │ │ │ ├── test_losses.py │ │ │ │ │ ├── test_mask_ops.py │ │ │ │ │ ├── test_nms.py │ │ │ │ │ ├── test_nms_rotated.py │ │ │ │ │ ├── test_roi_align.py │ │ │ │ │ └── test_roi_align_rotated.py │ │ │ │ ├── modeling │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_anchor_generator.py │ │ │ │ │ ├── test_backbone.py │ │ │ │ │ ├── test_box2box_transform.py │ │ │ │ │ ├── test_fast_rcnn.py │ │ │ │ │ ├── test_matcher.py │ │ │ │ │ ├── test_mmdet.py │ │ │ │ │ ├── test_model_e2e.py │ │ │ │ │ ├── test_roi_heads.py │ │ │ │ │ ├── test_roi_pooler.py │ │ │ │ │ └── test_rpn.py │ │ │ │ ├── structures │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_boxes.py │ │ │ │ │ ├── test_imagelist.py │ │ │ │ │ ├── test_instances.py │ │ │ │ │ ├── test_keypoints.py │ │ │ │ │ ├── test_masks.py │ │ │ │ │ └── test_rotated_boxes.py │ │ │ │ ├── test_checkpoint.py │ │ │ │ ├── test_engine.py │ │ │ │ ├── test_events.py │ │ │ │ ├── test_export_caffe2.py │ │ │ │ ├── test_export_onnx.py │ │ │ │ ├── test_export_torchscript.py │ │ │ │ ├── test_model_analysis.py │ │ │ │ ├── test_model_zoo.py │ │ │ │ ├── test_packaging.py │ │ │ │ ├── test_registry.py │ │ │ │ ├── test_scheduler.py │ │ │ │ ├── test_solver.py │ │ │ │ ├── test_visualizer.py │ │ │ │ ├── tracking │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_bbox_iou_tracker.py │ │ │ │ │ ├── test_hungarian_tracker.py │ │ │ │ │ ├── test_iou_weighted_hungarian_bbox_iou_tracker.py │ │ │ │ │ └── test_vanilla_hungarian_bbox_iou_tracker.py │ │ │ │ └── utils │ │ │ │ │ └── test_tensorboardx.py │ │ │ └── tools │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── analyze_model.py │ │ │ │ ├── benchmark.py │ │ │ │ ├── convert-torchvision-to-d2.py │ │ │ │ ├── deploy │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── README.md │ │ │ │ ├── export_model.py │ │ │ │ └── torchscript_mask_rcnn.cpp │ │ │ │ ├── lazyconfig_train_net.py │ │ │ │ ├── lightning_train_net.py │ │ │ │ ├── plain_train_net.py │ │ │ │ ├── train_net.py │ │ │ │ ├── visualize_data.py │ │ │ │ └── visualize_json_results.py │ │ ├── mamba-1p1p1 │ │ │ ├── .github │ │ │ │ └── workflows │ │ │ │ │ └── publish.yaml │ │ │ ├── .gitignore │ │ │ ├── .gitmodules │ │ │ ├── AUTHORS │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── assets │ │ │ │ └── selection.png │ │ │ ├── benchmarks │ │ │ │ └── benchmark_generation_mamba_simple.py │ │ │ ├── csrc │ │ │ │ └── selective_scan │ │ │ │ │ ├── reverse_scan.cuh │ │ │ │ │ ├── selective_scan.cpp │ │ │ │ │ ├── selective_scan.h │ │ │ │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ │ │ │ ├── selective_scan_bwd_bf16_real.cu │ │ │ │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ │ │ │ ├── selective_scan_bwd_fp16_real.cu │ │ │ │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ │ │ │ ├── selective_scan_bwd_fp32_real.cu │ │ │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ │ │ ├── selective_scan_common.h │ │ │ │ │ ├── selective_scan_fwd_bf16.cu │ │ │ │ │ ├── selective_scan_fwd_fp16.cu │ │ │ │ │ ├── selective_scan_fwd_fp32.cu │ │ │ │ │ ├── selective_scan_fwd_kernel.cuh │ │ │ │ │ ├── static_switch.h │ │ │ │ │ └── uninitialized_copy.cuh │ │ │ ├── evals │ │ │ │ └── lm_harness_eval.py │ │ │ ├── mamba_ssm │ │ │ │ ├── __init__.py │ │ │ │ ├── models │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config_mamba.py │ │ │ │ │ └── mixer_seq_simple.py │ │ │ │ ├── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── mamba_simple.py │ │ │ │ ├── ops │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── selective_scan_interface.py │ │ │ │ │ └── triton │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── layernorm.py │ │ │ │ │ │ └── selective_state_update.py │ │ │ │ └── utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── generation.py │ │ │ │ │ └── hf.py │ │ │ ├── setup.py │ │ │ └── tests │ │ │ │ └── ops │ │ │ │ ├── test_selective_scan.py │ │ │ │ └── triton │ │ │ │ └── test_selective_state_update.py │ │ ├── seg │ │ │ ├── README.md │ │ │ ├── backbone │ │ │ │ ├── __init__.py │ │ │ │ └── vim.py │ │ │ ├── configs │ │ │ │ ├── _base_ │ │ │ │ │ ├── datasets │ │ │ │ │ │ ├── ade20k.py │ │ │ │ │ │ ├── ade20k_1024x1024.py │ │ │ │ │ │ ├── ade20k_640x640.py │ │ │ │ │ │ └── ade20k_ms_eval.py │ │ │ │ │ ├── default_runtime.py │ │ │ │ │ ├── models │ │ │ │ │ │ ├── plainseg_vim.py │ │ │ │ │ │ ├── upernet_beit.py │ │ │ │ │ │ └── upernet_vim.py │ │ │ │ │ └── schedules │ │ │ │ │ │ ├── schedule_60k.py │ │ │ │ │ │ └── schedule_80k.py │ │ │ │ └── vim │ │ │ │ │ └── upernet │ │ │ │ │ ├── upernet_vim_small_24_512_slide_60k.py │ │ │ │ │ └── upernet_vim_tiny_24_512_slide_60k.py │ │ │ ├── mmcv_custom │ │ │ │ ├── __init__.py │ │ │ │ ├── apex_runner │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── apex_iter_based_runner.py │ │ │ │ │ ├── checkpoint.py │ │ │ │ │ └── optimizer.py │ │ │ │ ├── checkpoint.py │ │ │ │ ├── layer_decay_optimizer_constructor.py │ │ │ │ ├── resize_transform.py │ │ │ │ └── train_api.py │ │ │ ├── scripts │ │ │ │ ├── eval_vim_small_upernet.sh │ │ │ │ ├── eval_vim_tiny_upernet.sh │ │ │ │ ├── ft_vim_small_upernet.sh │ │ │ │ └── ft_vim_tiny_upernet.sh │ │ │ ├── seg-requirements.txt │ │ │ ├── test.py │ │ │ ├── tools │ │ │ │ └── analysis_tools │ │ │ │ │ ├── benchmark.py │ │ │ │ │ ├── benchmark_scripts.sh │ │ │ │ │ ├── deit-t-upernet.txt │ │ │ │ │ ├── plot_curve.ipynb │ │ │ │ │ └── vim-t-upernet.txt │ │ │ └── train.py │ │ └── vim │ │ │ ├── .gitignore │ │ │ ├── =0.9.0 │ │ │ ├── =1.8.0 │ │ │ ├── LICENSE │ │ │ ├── augment.py │ │ │ ├── calc_flops.py │ │ │ ├── datasets.py │ │ │ ├── engine.py │ │ │ ├── hubconf.py │ │ │ ├── losses.py │ │ │ ├── main.py │ │ │ ├── models │ │ │ ├── models_mamba.py │ │ │ ├── videomamba.py │ │ │ └── vim.py │ │ │ ├── rope.py │ │ │ ├── run_with_submitit.py │ │ │ ├── samplers.py │ │ │ ├── scripts │ │ │ └── ft-vim-t-random.sh │ │ │ ├── utils.py │ │ │ ├── vim_requirements.txt │ │ │ └── visualization.ipynb │ └── Static │ │ ├── .gitignore │ │ ├── =1.1.0 │ │ ├── LICENSE.txt │ │ ├── README.md │ │ ├── assets │ │ ├── vim_pipeline_v1.9.png │ │ └── vim_teaser_v1.7.png │ │ ├── det │ │ ├── .circleci │ │ │ ├── config.yml │ │ │ └── import-tests.sh │ │ ├── .clang-format │ │ ├── .flake8 │ │ ├── .gitignore │ │ ├── GETTING_STARTED.md │ │ ├── INSTALL.md │ │ ├── LICENSE │ │ ├── MODEL_ZOO.md │ │ ├── README.md │ │ ├── configs │ │ │ ├── Base-RCNN-C4.yaml │ │ │ ├── Base-RCNN-DilatedC5.yaml │ │ │ ├── Base-RCNN-FPN.yaml │ │ │ ├── Base-RetinaNet.yaml │ │ │ ├── COCO-Detection │ │ │ │ ├── fast_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── faster_rcnn_R_101_C4_3x.yaml │ │ │ │ ├── faster_rcnn_R_101_DC5_3x.yaml │ │ │ │ ├── faster_rcnn_R_101_FPN_3x.yaml │ │ │ │ ├── faster_rcnn_R_50_C4_1x.yaml │ │ │ │ ├── faster_rcnn_R_50_C4_3x.yaml │ │ │ │ ├── faster_rcnn_R_50_DC5_1x.yaml │ │ │ │ ├── faster_rcnn_R_50_DC5_3x.yaml │ │ │ │ ├── faster_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── faster_rcnn_R_50_FPN_3x.yaml │ │ │ │ ├── faster_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ ├── fcos_R_50_FPN_1x.py │ │ │ │ ├── retinanet_R_101_FPN_3x.yaml │ │ │ │ ├── retinanet_R_50_FPN_1x.py │ │ │ │ ├── retinanet_R_50_FPN_1x.yaml │ │ │ │ ├── retinanet_R_50_FPN_3x.yaml │ │ │ │ ├── rpn_R_50_C4_1x.yaml │ │ │ │ └── rpn_R_50_FPN_1x.yaml │ │ │ ├── COCO-InstanceSegmentation │ │ │ │ ├── mask_rcnn_R_101_C4_3x.yaml │ │ │ │ ├── mask_rcnn_R_101_DC5_3x.yaml │ │ │ │ ├── mask_rcnn_R_101_FPN_3x.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_1x.py │ │ │ │ ├── mask_rcnn_R_50_C4_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_3x.yaml │ │ │ │ ├── mask_rcnn_R_50_DC5_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_DC5_3x.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.py │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_giou.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_3x.yaml │ │ │ │ ├── mask_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_fpn_1x.py │ │ │ │ └── mask_rcnn_regnety_4gf_dds_fpn_1x.py │ │ │ ├── COCO-Keypoints │ │ │ │ ├── Base-Keypoint-RCNN-FPN.yaml │ │ │ │ ├── keypoint_rcnn_R_101_FPN_3x.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.py │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_3x.yaml │ │ │ │ └── keypoint_rcnn_X_101_32x8d_FPN_3x.yaml │ │ │ ├── COCO-PanopticSegmentation │ │ │ │ ├── Base-Panoptic-FPN.yaml │ │ │ │ ├── panoptic_fpn_R_101_3x.yaml │ │ │ │ ├── panoptic_fpn_R_50_1x.py │ │ │ │ ├── panoptic_fpn_R_50_1x.yaml │ │ │ │ └── panoptic_fpn_R_50_3x.yaml │ │ │ ├── Cityscapes │ │ │ │ └── mask_rcnn_R_50_FPN.yaml │ │ │ ├── Detectron1-Comparisons │ │ │ │ ├── README.md │ │ │ │ ├── faster_rcnn_R_50_FPN_noaug_1x.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_1x.yaml │ │ │ │ └── mask_rcnn_R_50_FPN_noaug_1x.yaml │ │ │ ├── LVISv0.5-InstanceSegmentation │ │ │ │ ├── mask_rcnn_R_101_FPN_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ └── mask_rcnn_X_101_32x8d_FPN_1x.yaml │ │ │ ├── LVISv1-InstanceSegmentation │ │ │ │ ├── mask_rcnn_R_101_FPN_1x.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ └── mask_rcnn_X_101_32x8d_FPN_1x.yaml │ │ │ ├── Misc │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_1x.yaml │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_3x.yaml │ │ │ │ ├── cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_cls_agnostic.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_1x_dconv_c3-c5.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_dconv_c3-c5.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_gn.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_syncbn.yaml │ │ │ │ ├── mmdet_mask_rcnn_R_50_FPN_1x.py │ │ │ │ ├── panoptic_fpn_R_101_dconv_cascade_gn_3x.yaml │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_3x_gn.yaml │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_9x_gn.yaml │ │ │ │ ├── scratch_mask_rcnn_R_50_FPN_9x_syncbn.yaml │ │ │ │ ├── semantic_R_50_FPN_1x.yaml │ │ │ │ └── torchvision_imagenet_R_50.py │ │ │ ├── PascalVOC-Detection │ │ │ │ ├── faster_rcnn_R_50_C4.yaml │ │ │ │ └── faster_rcnn_R_50_FPN.yaml │ │ │ ├── common │ │ │ │ ├── README.md │ │ │ │ ├── coco_schedule.py │ │ │ │ ├── data │ │ │ │ │ ├── coco.py │ │ │ │ │ ├── coco_keypoint.py │ │ │ │ │ ├── coco_panoptic_separated.py │ │ │ │ │ └── constants.py │ │ │ │ ├── models │ │ │ │ │ ├── cascade_rcnn.py │ │ │ │ │ ├── fcos.py │ │ │ │ │ ├── keypoint_rcnn_fpn.py │ │ │ │ │ ├── mask_rcnn_c4.py │ │ │ │ │ ├── mask_rcnn_fpn.py │ │ │ │ │ ├── mask_rcnn_vimdet.py │ │ │ │ │ ├── mask_rcnn_vitdet.py │ │ │ │ │ ├── panoptic_fpn.py │ │ │ │ │ └── retinanet.py │ │ │ │ ├── optim.py │ │ │ │ └── train.py │ │ │ ├── new_baselines │ │ │ │ ├── mask_rcnn_R_101_FPN_100ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_101_FPN_200ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_101_FPN_400ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_50_FPN_100ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_50_FPN_200ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_50_FPN_400ep_LSJ.py │ │ │ │ ├── mask_rcnn_R_50_FPN_50ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_100ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_200ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnetx_4gf_dds_FPN_400ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnety_4gf_dds_FPN_100ep_LSJ.py │ │ │ │ ├── mask_rcnn_regnety_4gf_dds_FPN_200ep_LSJ.py │ │ │ │ └── mask_rcnn_regnety_4gf_dds_FPN_400ep_LSJ.py │ │ │ └── quick_schedules │ │ │ │ ├── README.md │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── cascade_mask_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ ├── fast_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── fast_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_normalized_training_acc_test.yaml │ │ │ │ ├── keypoint_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_GCV_instant_test.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_inference_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_instant_test.yaml │ │ │ │ ├── mask_rcnn_R_50_C4_training_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_DC5_inference_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_pred_boxes_training_acc_test.yaml │ │ │ │ ├── mask_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ ├── panoptic_fpn_R_50_inference_acc_test.yaml │ │ │ │ ├── panoptic_fpn_R_50_instant_test.yaml │ │ │ │ ├── panoptic_fpn_R_50_training_acc_test.yaml │ │ │ │ ├── retinanet_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── retinanet_R_50_FPN_instant_test.yaml │ │ │ │ ├── rpn_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── rpn_R_50_FPN_instant_test.yaml │ │ │ │ ├── semantic_R_50_FPN_inference_acc_test.yaml │ │ │ │ ├── semantic_R_50_FPN_instant_test.yaml │ │ │ │ └── semantic_R_50_FPN_training_acc_test.yaml │ │ ├── datasets │ │ │ ├── README.md │ │ │ ├── prepare_ade20k_sem_seg.py │ │ │ ├── prepare_cocofied_lvis.py │ │ │ ├── prepare_for_tests.sh │ │ │ └── prepare_panoptic_fpn.py │ │ ├── demo │ │ │ ├── README.md │ │ │ ├── demo.py │ │ │ └── predictor.py │ │ ├── det-requirements.txt │ │ ├── detectron2 │ │ │ ├── __init__.py │ │ │ ├── checkpoint │ │ │ │ ├── __init__.py │ │ │ │ ├── c2_model_loading.py │ │ │ │ ├── catalog.py │ │ │ │ └── detection_checkpoint.py │ │ │ ├── config │ │ │ │ ├── __init__.py │ │ │ │ ├── compat.py │ │ │ │ ├── config.py │ │ │ │ ├── defaults.py │ │ │ │ ├── instantiate.py │ │ │ │ └── lazy.py │ │ │ ├── data │ │ │ │ ├── __init__.py │ │ │ │ ├── benchmark.py │ │ │ │ ├── build.py │ │ │ │ ├── catalog.py │ │ │ │ ├── common.py │ │ │ │ ├── dataset_mapper.py │ │ │ │ ├── datasets │ │ │ │ │ ├── README.md │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── builtin.py │ │ │ │ │ ├── builtin_meta.py │ │ │ │ │ ├── cityscapes.py │ │ │ │ │ ├── cityscapes_panoptic.py │ │ │ │ │ ├── coco.py │ │ │ │ │ ├── coco_panoptic.py │ │ │ │ │ ├── lvis.py │ │ │ │ │ ├── lvis_v0_5_categories.py │ │ │ │ │ ├── lvis_v1_categories.py │ │ │ │ │ ├── lvis_v1_category_image_count.py │ │ │ │ │ ├── pascal_voc.py │ │ │ │ │ └── register_coco.py │ │ │ │ ├── detection_utils.py │ │ │ │ ├── samplers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── distributed_sampler.py │ │ │ │ │ └── grouped_batch_sampler.py │ │ │ │ └── transforms │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── augmentation.py │ │ │ │ │ ├── augmentation_impl.py │ │ │ │ │ └── transform.py │ │ │ ├── engine │ │ │ │ ├── __init__.py │ │ │ │ ├── defaults.py │ │ │ │ ├── hooks.py │ │ │ │ ├── launch.py │ │ │ │ └── train_loop.py │ │ │ ├── evaluation │ │ │ │ ├── __init__.py │ │ │ │ ├── cityscapes_evaluation.py │ │ │ │ ├── coco_evaluation.py │ │ │ │ ├── evaluator.py │ │ │ │ ├── fast_eval_api.py │ │ │ │ ├── lvis_evaluation.py │ │ │ │ ├── panoptic_evaluation.py │ │ │ │ ├── pascal_voc_evaluation.py │ │ │ │ ├── rotated_coco_evaluation.py │ │ │ │ ├── sem_seg_evaluation.py │ │ │ │ └── testing.py │ │ │ ├── export │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── api.py │ │ │ │ ├── c10.py │ │ │ │ ├── caffe2_export.py │ │ │ │ ├── caffe2_inference.py │ │ │ │ ├── caffe2_modeling.py │ │ │ │ ├── caffe2_patch.py │ │ │ │ ├── flatten.py │ │ │ │ ├── shared.py │ │ │ │ ├── torchscript.py │ │ │ │ └── torchscript_patch.py │ │ │ ├── layers │ │ │ │ ├── __init__.py │ │ │ │ ├── aspp.py │ │ │ │ ├── batch_norm.py │ │ │ │ ├── blocks.py │ │ │ │ ├── csrc │ │ │ │ │ ├── README.md │ │ │ │ │ ├── ROIAlignRotated │ │ │ │ │ │ ├── ROIAlignRotated.h │ │ │ │ │ │ ├── ROIAlignRotated_cpu.cpp │ │ │ │ │ │ └── ROIAlignRotated_cuda.cu │ │ │ │ │ ├── box_iou_rotated │ │ │ │ │ │ ├── box_iou_rotated.h │ │ │ │ │ │ ├── box_iou_rotated_cpu.cpp │ │ │ │ │ │ ├── box_iou_rotated_cuda.cu │ │ │ │ │ │ └── box_iou_rotated_utils.h │ │ │ │ │ ├── cocoeval │ │ │ │ │ │ ├── cocoeval.cpp │ │ │ │ │ │ └── cocoeval.h │ │ │ │ │ ├── cuda_version.cu │ │ │ │ │ ├── deformable │ │ │ │ │ │ ├── deform_conv.h │ │ │ │ │ │ ├── deform_conv_cuda.cu │ │ │ │ │ │ └── deform_conv_cuda_kernel.cu │ │ │ │ │ ├── nms_rotated │ │ │ │ │ │ ├── nms_rotated.h │ │ │ │ │ │ ├── nms_rotated_cpu.cpp │ │ │ │ │ │ └── nms_rotated_cuda.cu │ │ │ │ │ └── vision.cpp │ │ │ │ ├── deform_conv.py │ │ │ │ ├── losses.py │ │ │ │ ├── mask_ops.py │ │ │ │ ├── nms.py │ │ │ │ ├── roi_align.py │ │ │ │ ├── roi_align_rotated.py │ │ │ │ ├── rotated_boxes.py │ │ │ │ ├── shape_spec.py │ │ │ │ └── wrappers.py │ │ │ ├── model_zoo │ │ │ │ ├── __init__.py │ │ │ │ └── model_zoo.py │ │ │ ├── modeling │ │ │ │ ├── __init__.py │ │ │ │ ├── anchor_generator.py │ │ │ │ ├── backbone │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── backbone.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── fpn.py │ │ │ │ │ ├── mvit.py │ │ │ │ │ ├── regnet.py │ │ │ │ │ ├── resnet.py │ │ │ │ │ ├── swin.py │ │ │ │ │ ├── utils.py │ │ │ │ │ ├── vim.py │ │ │ │ │ └── vit.py │ │ │ │ ├── box_regression.py │ │ │ │ ├── matcher.py │ │ │ │ ├── meta_arch │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── dense_detector.py │ │ │ │ │ ├── fcos.py │ │ │ │ │ ├── panoptic_fpn.py │ │ │ │ │ ├── rcnn.py │ │ │ │ │ ├── retinanet.py │ │ │ │ │ └── semantic_seg.py │ │ │ │ ├── mmdet_wrapper.py │ │ │ │ ├── poolers.py │ │ │ │ ├── postprocessing.py │ │ │ │ ├── proposal_generator │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── proposal_utils.py │ │ │ │ │ ├── rpn.py │ │ │ │ │ └── rrpn.py │ │ │ │ ├── roi_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── box_head.py │ │ │ │ │ ├── cascade_rcnn.py │ │ │ │ │ ├── fast_rcnn.py │ │ │ │ │ ├── keypoint_head.py │ │ │ │ │ ├── mask_head.py │ │ │ │ │ ├── roi_heads.py │ │ │ │ │ └── rotated_fast_rcnn.py │ │ │ │ ├── sampling.py │ │ │ │ └── test_time_augmentation.py │ │ │ ├── projects │ │ │ │ ├── README.md │ │ │ │ └── __init__.py │ │ │ ├── solver │ │ │ │ ├── __init__.py │ │ │ │ ├── build.py │ │ │ │ └── lr_scheduler.py │ │ │ ├── structures │ │ │ │ ├── __init__.py │ │ │ │ ├── boxes.py │ │ │ │ ├── image_list.py │ │ │ │ ├── instances.py │ │ │ │ ├── keypoints.py │ │ │ │ ├── masks.py │ │ │ │ └── rotated_boxes.py │ │ │ ├── tracking │ │ │ │ ├── __init__.py │ │ │ │ ├── base_tracker.py │ │ │ │ ├── bbox_iou_tracker.py │ │ │ │ ├── hungarian_tracker.py │ │ │ │ ├── iou_weighted_hungarian_bbox_iou_tracker.py │ │ │ │ ├── utils.py │ │ │ │ └── vanilla_hungarian_bbox_iou_tracker.py │ │ │ └── utils │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── analysis.py │ │ │ │ ├── collect_env.py │ │ │ │ ├── colormap.py │ │ │ │ ├── comm.py │ │ │ │ ├── develop.py │ │ │ │ ├── env.py │ │ │ │ ├── events.py │ │ │ │ ├── file_io.py │ │ │ │ ├── logger.py │ │ │ │ ├── memory.py │ │ │ │ ├── registry.py │ │ │ │ ├── serialize.py │ │ │ │ ├── testing.py │ │ │ │ ├── tracing.py │ │ │ │ ├── video_visualizer.py │ │ │ │ └── visualizer.py │ │ ├── dev │ │ │ ├── README.md │ │ │ ├── linter.sh │ │ │ ├── packaging │ │ │ │ ├── README.md │ │ │ │ ├── build_all_wheels.sh │ │ │ │ ├── build_wheel.sh │ │ │ │ ├── gen_install_table.py │ │ │ │ ├── gen_wheel_index.sh │ │ │ │ └── pkg_helpers.bash │ │ │ ├── parse_results.sh │ │ │ ├── run_inference_tests.sh │ │ │ └── run_instant_tests.sh │ │ ├── docker │ │ │ ├── Dockerfile │ │ │ ├── README.md │ │ │ ├── deploy.Dockerfile │ │ │ └── docker-compose.yml │ │ ├── docs │ │ │ ├── .gitignore │ │ │ ├── Makefile │ │ │ ├── README.md │ │ │ ├── _static │ │ │ │ └── css │ │ │ │ │ └── custom.css │ │ │ ├── conf.py │ │ │ ├── index.rst │ │ │ ├── modules │ │ │ │ ├── checkpoint.rst │ │ │ │ ├── config.rst │ │ │ │ ├── data.rst │ │ │ │ ├── data_transforms.rst │ │ │ │ ├── engine.rst │ │ │ │ ├── evaluation.rst │ │ │ │ ├── export.rst │ │ │ │ ├── fvcore.rst │ │ │ │ ├── index.rst │ │ │ │ ├── layers.rst │ │ │ │ ├── model_zoo.rst │ │ │ │ ├── modeling.rst │ │ │ │ ├── solver.rst │ │ │ │ ├── structures.rst │ │ │ │ └── utils.rst │ │ │ ├── notes │ │ │ │ ├── benchmarks.md │ │ │ │ ├── changelog.md │ │ │ │ ├── compatibility.md │ │ │ │ ├── contributing.md │ │ │ │ └── index.rst │ │ │ ├── requirements.txt │ │ │ └── tutorials │ │ │ │ ├── README.md │ │ │ │ ├── augmentation.md │ │ │ │ ├── builtin_datasets.md │ │ │ │ ├── configs.md │ │ │ │ ├── data_loading.md │ │ │ │ ├── datasets.md │ │ │ │ ├── deployment.md │ │ │ │ ├── evaluation.md │ │ │ │ ├── extend.md │ │ │ │ ├── getting_started.md │ │ │ │ ├── index.rst │ │ │ │ ├── install.md │ │ │ │ ├── lazyconfigs.md │ │ │ │ ├── models.md │ │ │ │ ├── training.md │ │ │ │ └── write-models.md │ │ ├── projects │ │ │ ├── DeepLab │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ └── Cityscapes-SemanticSegmentation │ │ │ │ │ │ ├── Base-DeepLabV3-OS16-Semantic.yaml │ │ │ │ │ │ ├── deeplab_v3_R_103_os16_mg124_poly_90k_bs16.yaml │ │ │ │ │ │ └── deeplab_v3_plus_R_103_os16_mg124_poly_90k_bs16.yaml │ │ │ │ ├── deeplab │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build_solver.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── loss.py │ │ │ │ │ ├── lr_scheduler.py │ │ │ │ │ ├── resnet.py │ │ │ │ │ └── semantic_seg.py │ │ │ │ └── train_net.py │ │ │ ├── DensePose │ │ │ │ ├── README.md │ │ │ │ ├── apply_net.py │ │ │ │ ├── configs │ │ │ │ │ ├── Base-DensePose-RCNN-FPN.yaml │ │ │ │ │ ├── HRNet │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w32_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w40_s1x.yaml │ │ │ │ │ │ └── densepose_rcnn_HRFPN_HRNet_w48_s1x.yaml │ │ │ │ │ ├── cse │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN-Human.yaml │ │ │ │ │ │ ├── Base-DensePose-RCNN-FPN.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_soft_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_soft_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_soft_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_CA_finetune_4k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_i2m_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_I0_finetune_m2m_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_16k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_4k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_animals_finetune_maskonly_24k.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_soft_chimps_finetune_4k.yaml │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_soft_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC1M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC1_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC2M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_WC2_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_DL_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC1M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC1_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC2M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_WC2_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_101_FPN_s1x_legacy.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC1M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC1_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC2M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_WC2_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2M_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x.yaml │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_s1x_legacy.yaml │ │ │ │ │ ├── evolution │ │ │ │ │ │ ├── Base-RCNN-FPN-Atop10P_CA.yaml │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA.yaml │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_coarsesegm.yaml │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_finesegm.yaml │ │ │ │ │ │ ├── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_uniform.yaml │ │ │ │ │ │ └── densepose_R_50_FPN_DL_WC1M_3x_Atop10P_CA_B_uv.yaml │ │ │ │ │ └── quick_schedules │ │ │ │ │ │ ├── cse │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_instant_test.yaml │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_soft_animals_finetune_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_HRFPN_HRNet_w32_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_DL_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_TTA_inference_acc_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC1_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_WC2_instant_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_inference_acc_test.yaml │ │ │ │ │ │ ├── densepose_rcnn_R_50_FPN_instant_test.yaml │ │ │ │ │ │ └── densepose_rcnn_R_50_FPN_training_acc_test.yaml │ │ │ │ ├── densepose │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── converters │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── base.py │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ ├── chart_output_hflip.py │ │ │ │ │ │ ├── chart_output_to_chart_result.py │ │ │ │ │ │ ├── hflip.py │ │ │ │ │ │ ├── segm_to_mask.py │ │ │ │ │ │ ├── to_chart_result.py │ │ │ │ │ │ └── to_mask.py │ │ │ │ │ ├── data │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── combined_loader.py │ │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ │ ├── datasets │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ ├── chimpnsee.py │ │ │ │ │ │ │ ├── coco.py │ │ │ │ │ │ │ ├── dataset_type.py │ │ │ │ │ │ │ └── lvis.py │ │ │ │ │ │ ├── image_list_dataset.py │ │ │ │ │ │ ├── inference_based_loader.py │ │ │ │ │ │ ├── meshes │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── builtin.py │ │ │ │ │ │ │ └── catalog.py │ │ │ │ │ │ ├── samplers │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── densepose_base.py │ │ │ │ │ │ │ ├── densepose_confidence_based.py │ │ │ │ │ │ │ ├── densepose_cse_base.py │ │ │ │ │ │ │ ├── densepose_cse_confidence_based.py │ │ │ │ │ │ │ ├── densepose_cse_uniform.py │ │ │ │ │ │ │ ├── densepose_uniform.py │ │ │ │ │ │ │ ├── mask_from_densepose.py │ │ │ │ │ │ │ └── prediction_to_gt.py │ │ │ │ │ │ ├── transform │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ └── image.py │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ └── video │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── frame_selector.py │ │ │ │ │ │ │ └── video_keyframe_dataset.py │ │ │ │ │ ├── engine │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── trainer.py │ │ │ │ │ ├── evaluation │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── d2_evaluator_adapter.py │ │ │ │ │ │ ├── densepose_coco_evaluation.py │ │ │ │ │ │ ├── evaluator.py │ │ │ │ │ │ ├── mesh_alignment_evaluator.py │ │ │ │ │ │ └── tensor_storage.py │ │ │ │ │ ├── modeling │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── build.py │ │ │ │ │ │ ├── confidence.py │ │ │ │ │ │ ├── cse │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── embedder.py │ │ │ │ │ │ │ ├── utils.py │ │ │ │ │ │ │ ├── vertex_direct_embedder.py │ │ │ │ │ │ │ └── vertex_feature_embedder.py │ │ │ │ │ │ ├── densepose_checkpoint.py │ │ │ │ │ │ ├── filter.py │ │ │ │ │ │ ├── hrfpn.py │ │ │ │ │ │ ├── hrnet.py │ │ │ │ │ │ ├── inference.py │ │ │ │ │ │ ├── losses │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ ├── chart_with_confidences.py │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ ├── cycle_pix2shape.py │ │ │ │ │ │ │ ├── cycle_shape2shape.py │ │ │ │ │ │ │ ├── embed.py │ │ │ │ │ │ │ ├── embed_utils.py │ │ │ │ │ │ │ ├── mask.py │ │ │ │ │ │ │ ├── mask_or_segm.py │ │ │ │ │ │ │ ├── registry.py │ │ │ │ │ │ │ ├── segm.py │ │ │ │ │ │ │ ├── soft_embed.py │ │ │ │ │ │ │ └── utils.py │ │ │ │ │ │ ├── predictors │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ │ ├── chart_confidence.py │ │ │ │ │ │ │ ├── chart_with_confidence.py │ │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ │ ├── cse_confidence.py │ │ │ │ │ │ │ ├── cse_with_confidence.py │ │ │ │ │ │ │ └── registry.py │ │ │ │ │ │ ├── roi_heads │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── deeplab.py │ │ │ │ │ │ │ ├── registry.py │ │ │ │ │ │ │ ├── roi_head.py │ │ │ │ │ │ │ └── v1convx.py │ │ │ │ │ │ ├── test_time_augmentation.py │ │ │ │ │ │ └── utils.py │ │ │ │ │ ├── structures │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── chart.py │ │ │ │ │ │ ├── chart_confidence.py │ │ │ │ │ │ ├── chart_result.py │ │ │ │ │ │ ├── cse.py │ │ │ │ │ │ ├── cse_confidence.py │ │ │ │ │ │ ├── data_relative.py │ │ │ │ │ │ ├── list.py │ │ │ │ │ │ ├── mesh.py │ │ │ │ │ │ └── transform_data.py │ │ │ │ │ ├── utils │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── dbhelper.py │ │ │ │ │ │ ├── logger.py │ │ │ │ │ │ └── transform.py │ │ │ │ │ └── vis │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── base.py │ │ │ │ │ │ ├── bounding_box.py │ │ │ │ │ │ ├── densepose_data_points.py │ │ │ │ │ │ ├── densepose_outputs_iuv.py │ │ │ │ │ │ ├── densepose_outputs_vertex.py │ │ │ │ │ │ ├── densepose_results.py │ │ │ │ │ │ ├── densepose_results_textures.py │ │ │ │ │ │ └── extractor.py │ │ │ │ ├── dev │ │ │ │ │ ├── README.md │ │ │ │ │ ├── run_inference_tests.sh │ │ │ │ │ └── run_instant_tests.sh │ │ │ │ ├── doc │ │ │ │ │ ├── BOOTSTRAPPING_PIPELINE.md │ │ │ │ │ ├── DENSEPOSE_CSE.md │ │ │ │ │ ├── DENSEPOSE_DATASETS.md │ │ │ │ │ ├── DENSEPOSE_IUV.md │ │ │ │ │ ├── GETTING_STARTED.md │ │ │ │ │ ├── RELEASE_2020_04.md │ │ │ │ │ ├── RELEASE_2021_03.md │ │ │ │ │ ├── RELEASE_2021_06.md │ │ │ │ │ ├── TOOL_APPLY_NET.md │ │ │ │ │ └── TOOL_QUERY_DB.md │ │ │ │ ├── query_db.py │ │ │ │ ├── setup.py │ │ │ │ ├── tests │ │ │ │ │ ├── common.py │ │ │ │ │ ├── test_chart_based_annotations_accumulator.py │ │ │ │ │ ├── test_combine_data_loader.py │ │ │ │ │ ├── test_cse_annotations_accumulator.py │ │ │ │ │ ├── test_dataset_loaded_annotations.py │ │ │ │ │ ├── test_frame_selector.py │ │ │ │ │ ├── test_image_list_dataset.py │ │ │ │ │ ├── test_image_resize_transform.py │ │ │ │ │ ├── test_model_e2e.py │ │ │ │ │ ├── test_setup.py │ │ │ │ │ ├── test_structures.py │ │ │ │ │ ├── test_tensor_storage.py │ │ │ │ │ └── test_video_keyframe_dataset.py │ │ │ │ └── train_net.py │ │ │ ├── MViTv2 │ │ │ │ ├── README.md │ │ │ │ └── configs │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_3x.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_3x.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_lsj_3x.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_lsj_50ep.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_s_3x.py │ │ │ │ │ ├── cascade_mask_rcnn_mvitv2_t_3x.py │ │ │ │ │ ├── common │ │ │ │ │ ├── coco_loader.py │ │ │ │ │ └── coco_loader_lsj.py │ │ │ │ │ └── mask_rcnn_mvitv2_t_3x.py │ │ │ ├── Panoptic-DeepLab │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── COCO-PanopticSegmentation │ │ │ │ │ │ └── panoptic_deeplab_R_52_os16_mg124_poly_200k_bs64_crop_640_640_coco_dsconv.yaml │ │ │ │ │ └── Cityscapes-PanopticSegmentation │ │ │ │ │ │ ├── Base-PanopticDeepLab-OS16.yaml │ │ │ │ │ │ ├── panoptic_deeplab_R_52_os16_mg124_poly_90k_bs32_crop_512_1024.yaml │ │ │ │ │ │ └── panoptic_deeplab_R_52_os16_mg124_poly_90k_bs32_crop_512_1024_dsconv.yaml │ │ │ │ ├── panoptic_deeplab │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ ├── panoptic_seg.py │ │ │ │ │ ├── post_processing.py │ │ │ │ │ └── target_generator.py │ │ │ │ └── train_net.py │ │ │ ├── PointRend │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── InstanceSegmentation │ │ │ │ │ │ ├── Base-Implicit-PointRend.yaml │ │ │ │ │ │ ├── Base-PointRend-RCNN-FPN.yaml │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_1x_coco.yaml │ │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_3x_coco.yaml │ │ │ │ │ │ ├── pointrend_rcnn_R_101_FPN_3x_coco.yaml │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_1x_cityscapes.yaml │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_1x_coco.yaml │ │ │ │ │ │ ├── pointrend_rcnn_R_50_FPN_3x_coco.yaml │ │ │ │ │ │ └── pointrend_rcnn_X_101_32x8d_FPN_3x_coco.yaml │ │ │ │ │ └── SemanticSegmentation │ │ │ │ │ │ ├── Base-PointRend-Semantic-FPN.yaml │ │ │ │ │ │ └── pointrend_semantic_R_101_FPN_1x_cityscapes.yaml │ │ │ │ ├── point_rend │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── color_augmentation.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── mask_head.py │ │ │ │ │ ├── point_features.py │ │ │ │ │ ├── point_head.py │ │ │ │ │ ├── roi_heads.py │ │ │ │ │ └── semantic_seg.py │ │ │ │ └── train_net.py │ │ │ ├── PointSup │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── implicit_pointrend_R_50_FPN_3x_point_sup_point_aug_coco.yaml │ │ │ │ │ ├── mask_rcnn_R_50_FPN_3x_point_sup_coco.yaml │ │ │ │ │ └── mask_rcnn_R_50_FPN_3x_point_sup_point_aug_coco.yaml │ │ │ │ ├── point_sup │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── dataset_mapper.py │ │ │ │ │ ├── detection_utils.py │ │ │ │ │ ├── mask_head.py │ │ │ │ │ ├── point_utils.py │ │ │ │ │ └── register_point_annotations.py │ │ │ │ ├── tools │ │ │ │ │ └── prepare_coco_point_annotations_without_masks.py │ │ │ │ └── train_net.py │ │ │ ├── README.md │ │ │ ├── Rethinking-BatchNorm │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── mask_rcnn_BNhead.py │ │ │ │ │ ├── mask_rcnn_BNhead_batch_stats.py │ │ │ │ │ ├── mask_rcnn_BNhead_shuffle.py │ │ │ │ │ ├── mask_rcnn_SyncBNhead.py │ │ │ │ │ ├── retinanet_SyncBNhead.py │ │ │ │ │ └── retinanet_SyncBNhead_SharedTraining.py │ │ │ │ └── retinanet-eval-domain-specific.py │ │ │ ├── TensorMask │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── Base-TensorMask.yaml │ │ │ │ │ ├── tensormask_R_50_FPN_1x.yaml │ │ │ │ │ └── tensormask_R_50_FPN_6x.yaml │ │ │ │ ├── setup.py │ │ │ │ ├── tensormask │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── arch.py │ │ │ │ │ ├── config.py │ │ │ │ │ └── layers │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── csrc │ │ │ │ │ │ ├── SwapAlign2Nat │ │ │ │ │ │ │ ├── SwapAlign2Nat.h │ │ │ │ │ │ │ └── SwapAlign2Nat_cuda.cu │ │ │ │ │ │ └── vision.cpp │ │ │ │ │ │ └── swap_align2nat.py │ │ │ │ ├── tests │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── test_swap_align2nat.py │ │ │ │ └── train_net.py │ │ │ ├── TridentNet │ │ │ │ ├── README.md │ │ │ │ ├── configs │ │ │ │ │ ├── Base-TridentNet-Fast-C4.yaml │ │ │ │ │ ├── tridentnet_fast_R_101_C4_3x.yaml │ │ │ │ │ ├── tridentnet_fast_R_50_C4_1x.yaml │ │ │ │ │ └── tridentnet_fast_R_50_C4_3x.yaml │ │ │ │ ├── train_net.py │ │ │ │ └── tridentnet │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── trident_backbone.py │ │ │ │ │ ├── trident_conv.py │ │ │ │ │ ├── trident_rcnn.py │ │ │ │ │ └── trident_rpn.py │ │ │ └── ViTDet │ │ │ │ ├── README.md │ │ │ │ └── configs │ │ │ │ ├── COCO │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_100ep.py │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_36ep.py │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_swin_b_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_swin_l_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_vimdet_b_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vimdet_s_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vimdet_t_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vimdet_t_100ep_adj1.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_b_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_h_75ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_l_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_s_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_t_100ep.py │ │ │ │ ├── lite_cascade_mask_rcnn_vimdet_t_100ep.py │ │ │ │ ├── mask_rcnn_vimdet_b_100ep.py │ │ │ │ ├── mask_rcnn_vitdet_b_100ep.py │ │ │ │ ├── mask_rcnn_vitdet_h_75ep.py │ │ │ │ ├── mask_rcnn_vitdet_l_100ep.py │ │ │ │ └── mask_rcnn_vitdet_t_100ep.py │ │ │ │ ├── LVIS │ │ │ │ ├── cascade_mask_rcnn_mvitv2_b_in21k_100ep.py │ │ │ │ ├── cascade_mask_rcnn_mvitv2_h_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_mvitv2_l_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_swin_b_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_swin_l_in21k_50ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_b_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_h_100ep.py │ │ │ │ ├── cascade_mask_rcnn_vitdet_l_100ep.py │ │ │ │ ├── mask_rcnn_vitdet_b_100ep.py │ │ │ │ ├── mask_rcnn_vitdet_h_100ep.py │ │ │ │ └── mask_rcnn_vitdet_l_100ep.py │ │ │ │ └── common │ │ │ │ └── coco_loader_lsj.py │ │ ├── scripts │ │ │ ├── eval_vim_tiny_vimdet.sh │ │ │ └── ft_vim_tiny_vimdet.sh │ │ ├── setup.cfg │ │ ├── setup.py │ │ ├── tests │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── config │ │ │ │ ├── dir1 │ │ │ │ │ ├── bad_import.py │ │ │ │ │ ├── bad_import2.py │ │ │ │ │ ├── dir1_a.py │ │ │ │ │ ├── dir1_b.py │ │ │ │ │ └── load_rel.py │ │ │ │ ├── root_cfg.py │ │ │ │ ├── test_instantiate_config.py │ │ │ │ ├── test_lazy_config.py │ │ │ │ └── test_yacs_config.py │ │ │ ├── data │ │ │ │ ├── __init__.py │ │ │ │ ├── test_coco.py │ │ │ │ ├── test_coco_evaluation.py │ │ │ │ ├── test_dataset.py │ │ │ │ ├── test_detection_utils.py │ │ │ │ ├── test_rotation_transform.py │ │ │ │ ├── test_sampler.py │ │ │ │ └── test_transforms.py │ │ │ ├── export │ │ │ │ └── test_c10.py │ │ │ ├── layers │ │ │ │ ├── __init__.py │ │ │ │ ├── test_blocks.py │ │ │ │ ├── test_deformable.py │ │ │ │ ├── test_losses.py │ │ │ │ ├── test_mask_ops.py │ │ │ │ ├── test_nms.py │ │ │ │ ├── test_nms_rotated.py │ │ │ │ ├── test_roi_align.py │ │ │ │ └── test_roi_align_rotated.py │ │ │ ├── modeling │ │ │ │ ├── __init__.py │ │ │ │ ├── test_anchor_generator.py │ │ │ │ ├── test_backbone.py │ │ │ │ ├── test_box2box_transform.py │ │ │ │ ├── test_fast_rcnn.py │ │ │ │ ├── test_matcher.py │ │ │ │ ├── test_mmdet.py │ │ │ │ ├── test_model_e2e.py │ │ │ │ ├── test_roi_heads.py │ │ │ │ ├── test_roi_pooler.py │ │ │ │ └── test_rpn.py │ │ │ ├── structures │ │ │ │ ├── __init__.py │ │ │ │ ├── test_boxes.py │ │ │ │ ├── test_imagelist.py │ │ │ │ ├── test_instances.py │ │ │ │ ├── test_keypoints.py │ │ │ │ ├── test_masks.py │ │ │ │ └── test_rotated_boxes.py │ │ │ ├── test_checkpoint.py │ │ │ ├── test_engine.py │ │ │ ├── test_events.py │ │ │ ├── test_export_caffe2.py │ │ │ ├── test_export_onnx.py │ │ │ ├── test_export_torchscript.py │ │ │ ├── test_model_analysis.py │ │ │ ├── test_model_zoo.py │ │ │ ├── test_packaging.py │ │ │ ├── test_registry.py │ │ │ ├── test_scheduler.py │ │ │ ├── test_solver.py │ │ │ ├── test_visualizer.py │ │ │ ├── tracking │ │ │ │ ├── __init__.py │ │ │ │ ├── test_bbox_iou_tracker.py │ │ │ │ ├── test_hungarian_tracker.py │ │ │ │ ├── test_iou_weighted_hungarian_bbox_iou_tracker.py │ │ │ │ └── test_vanilla_hungarian_bbox_iou_tracker.py │ │ │ └── utils │ │ │ │ └── test_tensorboardx.py │ │ └── tools │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── analyze_model.py │ │ │ ├── benchmark.py │ │ │ ├── convert-torchvision-to-d2.py │ │ │ ├── deploy │ │ │ ├── CMakeLists.txt │ │ │ ├── README.md │ │ │ ├── export_model.py │ │ │ └── torchscript_mask_rcnn.cpp │ │ │ ├── lazyconfig_train_net.py │ │ │ ├── lightning_train_net.py │ │ │ ├── plain_train_net.py │ │ │ ├── train_net.py │ │ │ ├── visualize_data.py │ │ │ └── visualize_json_results.py │ │ ├── mamba-1p1p1 │ │ ├── .github │ │ │ └── workflows │ │ │ │ └── publish.yaml │ │ ├── .gitignore │ │ ├── .gitmodules │ │ ├── AUTHORS │ │ ├── LICENSE │ │ ├── README.md │ │ ├── assets │ │ │ └── selection.png │ │ ├── benchmarks │ │ │ └── benchmark_generation_mamba_simple.py │ │ ├── csrc │ │ │ └── selective_scan │ │ │ │ ├── reverse_scan.cuh │ │ │ │ ├── selective_scan.cpp │ │ │ │ ├── selective_scan.h │ │ │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ │ │ ├── selective_scan_bwd_bf16_real.cu │ │ │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ │ │ ├── selective_scan_bwd_fp16_real.cu │ │ │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ │ │ ├── selective_scan_bwd_fp32_real.cu │ │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ │ ├── selective_scan_common.h │ │ │ │ ├── selective_scan_fwd_bf16.cu │ │ │ │ ├── selective_scan_fwd_fp16.cu │ │ │ │ ├── selective_scan_fwd_fp32.cu │ │ │ │ ├── selective_scan_fwd_kernel.cuh │ │ │ │ ├── static_switch.h │ │ │ │ └── uninitialized_copy.cuh │ │ ├── evals │ │ │ └── lm_harness_eval.py │ │ ├── mamba_ssm │ │ │ ├── __init__.py │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ ├── config_mamba.py │ │ │ │ └── mixer_seq_simple.py │ │ │ ├── modules │ │ │ │ ├── __init__.py │ │ │ │ └── mamba_simple.py │ │ │ ├── ops │ │ │ │ ├── __init__.py │ │ │ │ ├── selective_scan_interface.py │ │ │ │ └── triton │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── layernorm.py │ │ │ │ │ └── selective_state_update.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── generation.py │ │ │ │ └── hf.py │ │ ├── setup.py │ │ └── tests │ │ │ └── ops │ │ │ ├── test_selective_scan.py │ │ │ └── triton │ │ │ └── test_selective_state_update.py │ │ ├── seg │ │ ├── README.md │ │ ├── backbone │ │ │ ├── __init__.py │ │ │ └── vim.py │ │ ├── configs │ │ │ ├── _base_ │ │ │ │ ├── datasets │ │ │ │ │ ├── ade20k.py │ │ │ │ │ ├── ade20k_1024x1024.py │ │ │ │ │ ├── ade20k_640x640.py │ │ │ │ │ └── ade20k_ms_eval.py │ │ │ │ ├── default_runtime.py │ │ │ │ ├── models │ │ │ │ │ ├── plainseg_vim.py │ │ │ │ │ ├── upernet_beit.py │ │ │ │ │ └── upernet_vim.py │ │ │ │ └── schedules │ │ │ │ │ ├── schedule_60k.py │ │ │ │ │ └── schedule_80k.py │ │ │ └── vim │ │ │ │ └── upernet │ │ │ │ ├── upernet_vim_small_24_512_slide_60k.py │ │ │ │ └── upernet_vim_tiny_24_512_slide_60k.py │ │ ├── mmcv_custom │ │ │ ├── __init__.py │ │ │ ├── apex_runner │ │ │ │ ├── __init__.py │ │ │ │ ├── apex_iter_based_runner.py │ │ │ │ ├── checkpoint.py │ │ │ │ └── optimizer.py │ │ │ ├── checkpoint.py │ │ │ ├── layer_decay_optimizer_constructor.py │ │ │ ├── resize_transform.py │ │ │ └── train_api.py │ │ ├── scripts │ │ │ ├── eval_vim_small_upernet.sh │ │ │ ├── eval_vim_tiny_upernet.sh │ │ │ ├── ft_vim_small_upernet.sh │ │ │ └── ft_vim_tiny_upernet.sh │ │ ├── seg-requirements.txt │ │ ├── test.py │ │ ├── tools │ │ │ └── analysis_tools │ │ │ │ ├── benchmark.py │ │ │ │ ├── benchmark_scripts.sh │ │ │ │ ├── deit-t-upernet.txt │ │ │ │ ├── plot_curve.ipynb │ │ │ │ └── vim-t-upernet.txt │ │ └── train.py │ │ └── vim │ │ ├── .gitignore │ │ ├── =0.9.0 │ │ ├── =1.8.0 │ │ ├── LICENSE │ │ ├── augment.py │ │ ├── calc_flops.py │ │ ├── datasets.py │ │ ├── engine.py │ │ ├── hubconf.py │ │ ├── losses.py │ │ ├── main.py │ │ ├── models │ │ ├── models_mamba.py │ │ ├── videomamba.py │ │ └── vim.py │ │ ├── rope.py │ │ ├── run_with_submitit.py │ │ ├── samplers.py │ │ ├── scripts │ │ └── ft-vim-t-static.sh │ │ ├── utils.py │ │ ├── vim_requirements.txt │ │ └── visualization.ipynb ├── Token_Only │ ├── .gitignore │ ├── mamba-1p1p1 │ │ ├── AUTHORS │ │ ├── LICENSE │ │ ├── README.md │ │ ├── assets │ │ │ └── selection.png │ │ ├── benchmarks │ │ │ └── benchmark_generation_mamba_simple.py │ │ ├── csrc │ │ │ └── selective_scan │ │ │ │ ├── reverse_scan.cuh │ │ │ │ ├── selective_scan.cpp │ │ │ │ ├── selective_scan.h │ │ │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ │ │ ├── selective_scan_bwd_bf16_real.cu │ │ │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ │ │ ├── selective_scan_bwd_fp16_real.cu │ │ │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ │ │ ├── selective_scan_bwd_fp32_real.cu │ │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ │ ├── selective_scan_common.h │ │ │ │ ├── selective_scan_fwd_bf16.cu │ │ │ │ ├── selective_scan_fwd_fp16.cu │ │ │ │ ├── selective_scan_fwd_fp32.cu │ │ │ │ ├── selective_scan_fwd_kernel.cuh │ │ │ │ ├── static_switch.h │ │ │ │ └── uninitialized_copy.cuh │ │ ├── evals │ │ │ └── lm_harness_eval.py │ │ ├── mamba_ssm │ │ │ ├── __init__.py │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ ├── config_mamba.py │ │ │ │ └── mixer_seq_simple.py │ │ │ ├── modules │ │ │ │ ├── __init__.py │ │ │ │ └── mamba_simple.py │ │ │ ├── ops │ │ │ │ ├── __init__.py │ │ │ │ ├── selective_scan_interface.py │ │ │ │ └── triton │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── layernorm.py │ │ │ │ │ └── selective_state_update.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── generation.py │ │ │ │ └── hf.py │ │ ├── setup.py │ │ └── tests │ │ │ └── ops │ │ │ ├── test_selective_scan.py │ │ │ └── triton │ │ │ └── test_selective_state_update.py │ └── vim │ │ ├── LICENSE │ │ ├── augment.py │ │ ├── calc_flops.py │ │ ├── calc_throughput.py │ │ ├── datasets.py │ │ ├── engine.py │ │ ├── hubconf.py │ │ ├── losses.py │ │ ├── main.py │ │ ├── models │ │ ├── embedding.py │ │ ├── mamba_block.py │ │ ├── predictor.py │ │ └── vim_token_only.py │ │ ├── rope.py │ │ ├── samplers.py │ │ ├── scripts │ │ ├── vim-b.sh │ │ ├── vim-s.sh │ │ └── vim-t.sh │ │ ├── utils.py │ │ └── vim_requirements.txt └── readme.md ├── Asset └── pipeline.png ├── Image_Classification ├── MambaReg │ ├── causal-conv1d │ │ ├── AUTHORS │ │ ├── LICENSE │ │ ├── README.md │ │ ├── causal_conv1d.egg-info │ │ │ ├── PKG-INFO │ │ │ ├── SOURCES.txt │ │ │ ├── dependency_links.txt │ │ │ ├── requires.txt │ │ │ └── top_level.txt │ │ ├── causal_conv1d │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-310.pyc │ │ │ │ └── causal_conv1d_interface.cpython-310.pyc │ │ │ └── causal_conv1d_interface.py │ │ ├── csrc │ │ │ ├── causal_conv1d.cpp │ │ │ ├── causal_conv1d.h │ │ │ ├── causal_conv1d_bwd.cu │ │ │ ├── causal_conv1d_common.h │ │ │ ├── causal_conv1d_fwd.cu │ │ │ ├── causal_conv1d_update.cu │ │ │ └── static_switch.h │ │ ├── setup.py │ │ └── tests │ │ │ └── test_causal_conv1d.py │ ├── mamba-1p1p1 │ │ ├── AUTHORS │ │ ├── LICENSE │ │ ├── README.md │ │ ├── assets │ │ │ └── selection.png │ │ ├── benchmarks │ │ │ └── benchmark_generation_mamba_simple.py │ │ ├── csrc │ │ │ └── selective_scan │ │ │ │ ├── reverse_scan.cuh │ │ │ │ ├── selective_scan.cpp │ │ │ │ ├── selective_scan.h │ │ │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ │ │ ├── selective_scan_bwd_bf16_real.cu │ │ │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ │ │ ├── selective_scan_bwd_fp16_real.cu │ │ │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ │ │ ├── selective_scan_bwd_fp32_real.cu │ │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ │ ├── selective_scan_common.h │ │ │ │ ├── selective_scan_fwd_bf16.cu │ │ │ │ ├── selective_scan_fwd_fp16.cu │ │ │ │ ├── selective_scan_fwd_fp32.cu │ │ │ │ ├── selective_scan_fwd_kernel.cuh │ │ │ │ ├── static_switch.h │ │ │ │ └── uninitialized_copy.cuh │ │ ├── evals │ │ │ └── lm_harness_eval.py │ │ ├── mamba_ssm.egg-info │ │ │ ├── PKG-INFO │ │ │ ├── SOURCES.txt │ │ │ ├── dependency_links.txt │ │ │ ├── requires.txt │ │ │ └── top_level.txt │ │ ├── mamba_ssm │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ └── __init__.cpython-310.pyc │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ ├── __pycache__ │ │ │ │ │ ├── __init__.cpython-310.pyc │ │ │ │ │ ├── config_mamba.cpython-310.pyc │ │ │ │ │ └── mixer_seq_simple.cpython-310.pyc │ │ │ │ ├── config_mamba.py │ │ │ │ └── mixer_seq_simple.py │ │ │ ├── modules │ │ │ │ ├── .ipynb_checkpoints │ │ │ │ │ └── mamba_simple-checkpoint.py │ │ │ │ ├── __init__.py │ │ │ │ ├── __pycache__ │ │ │ │ │ ├── __init__.cpython-310.pyc │ │ │ │ │ └── mamba_simple.cpython-310.pyc │ │ │ │ └── mamba_simple.py │ │ │ ├── ops │ │ │ │ ├── __init__.py │ │ │ │ ├── __pycache__ │ │ │ │ │ ├── __init__.cpython-310.pyc │ │ │ │ │ └── selective_scan_interface.cpython-310.pyc │ │ │ │ ├── selective_scan_interface.py │ │ │ │ └── triton │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── __pycache__ │ │ │ │ │ ├── __init__.cpython-310.pyc │ │ │ │ │ ├── layernorm.cpython-310.pyc │ │ │ │ │ └── selective_state_update.cpython-310.pyc │ │ │ │ │ ├── layernorm.py │ │ │ │ │ └── selective_state_update.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-310.pyc │ │ │ │ ├── generation.cpython-310.pyc │ │ │ │ └── hf.cpython-310.pyc │ │ │ │ ├── generation.py │ │ │ │ └── hf.py │ │ ├── setup.py │ │ └── tests │ │ │ └── ops │ │ │ ├── test_selective_scan.py │ │ │ └── triton │ │ │ └── test_selective_state_update.py │ └── vim │ │ ├── .ipynb_checkpoints │ │ └── main-checkpoint.py │ │ ├── LICENSE │ │ ├── __pycache__ │ │ ├── augment.cpython-310.pyc │ │ ├── calc_flops.cpython-310.pyc │ │ ├── datasets.cpython-310.pyc │ │ ├── engine.cpython-310.pyc │ │ ├── losses.cpython-310.pyc │ │ ├── rope.cpython-310.pyc │ │ ├── samplers.cpython-310.pyc │ │ └── utils.cpython-310.pyc │ │ ├── augment.py │ │ ├── calc_flops.py │ │ ├── datasets.py │ │ ├── engine.py │ │ ├── hubconf.py │ │ ├── losses.py │ │ ├── main.py │ │ ├── mlruns │ │ └── 0 │ │ │ ├── 029766d5a88749ad83a0cb417b2b0bae │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── 24fb7b6a0f7e4880bab6c032bd77e973 │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── n_parameters │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── 24fcf6845c004274bd681e7d3851c721 │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── 2bb5ad6028374f0bb8f1b97c9a55a60e │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── n_parameters │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── 3ab1b30b4ed64e709f0333313df49100 │ │ │ ├── meta.yaml │ │ │ ├── metrics │ │ │ │ ├── epoch │ │ │ │ ├── n_parameters │ │ │ │ ├── test_acc1 │ │ │ │ ├── test_acc5 │ │ │ │ ├── test_block_keep_ratio │ │ │ │ ├── test_loss │ │ │ │ ├── train_block_keep_ratio │ │ │ │ ├── train_block_pruning_loss │ │ │ │ ├── train_cls_kl_loss │ │ │ │ ├── train_cls_loss │ │ │ │ ├── train_loss │ │ │ │ ├── train_lr │ │ │ │ ├── train_token_kl_loss │ │ │ │ └── train_token_pruning_loss │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── n_parameters │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── 5f3fa3b2ba5c4603ba890982f3e7daa0 │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── 66460dbc9d1f4ae2b174a274e1213147 │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── 87d8e3915a02419eaf8c830247982552 │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── 9bc80b4c33f647dc98b31985835ec501 │ │ │ ├── meta.yaml │ │ │ ├── metrics │ │ │ │ ├── epoch │ │ │ │ ├── n_parameters │ │ │ │ ├── test_acc1 │ │ │ │ ├── test_acc5 │ │ │ │ ├── test_block_keep_ratio │ │ │ │ ├── test_loss │ │ │ │ ├── train_block_keep_ratio │ │ │ │ ├── train_block_pruning_loss │ │ │ │ ├── train_cls_kl_loss │ │ │ │ ├── train_cls_loss │ │ │ │ ├── train_loss │ │ │ │ ├── train_lr │ │ │ │ ├── train_token_kl_loss │ │ │ │ └── train_token_pruning_loss │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── n_parameters │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── 9e3fd56185184925bb8823ce08931f33 │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── b41b98c70a4544a699ff1b76a77e7ffe │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── n_parameters │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── bb60d4d2bd004e2cb692dee5a73028b5 │ │ │ ├── meta.yaml │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ ├── c46388690df848a481bc263fb15306e4 │ │ │ ├── meta.yaml │ │ │ ├── metrics │ │ │ │ ├── epoch │ │ │ │ ├── n_parameters │ │ │ │ ├── test_acc1 │ │ │ │ ├── test_acc5 │ │ │ │ ├── test_block_keep_ratio │ │ │ │ ├── test_loss │ │ │ │ ├── train_block_keep_ratio │ │ │ │ ├── train_block_pruning_loss │ │ │ │ ├── train_cls_kl_loss │ │ │ │ ├── train_cls_loss │ │ │ │ ├── train_loss │ │ │ │ ├── train_lr │ │ │ │ ├── train_token_kl_loss │ │ │ │ └── train_token_pruning_loss │ │ │ ├── params │ │ │ │ ├── ThreeAugment │ │ │ │ ├── aa │ │ │ │ ├── attn_only │ │ │ │ ├── batch_size │ │ │ │ ├── bce_loss │ │ │ │ ├── block_keep_ratio │ │ │ │ ├── block_pruning_weight │ │ │ │ ├── clip_grad │ │ │ │ ├── color_jitter │ │ │ │ ├── cooldown_epochs │ │ │ │ ├── cosub │ │ │ │ ├── cutmix │ │ │ │ ├── cutmix_minmax │ │ │ │ ├── data_path │ │ │ │ ├── data_set │ │ │ │ ├── decay_epochs │ │ │ │ ├── decay_rate │ │ │ │ ├── device │ │ │ │ ├── dist_backend │ │ │ │ ├── dist_eval │ │ │ │ ├── dist_url │ │ │ │ ├── distillation_alpha │ │ │ │ ├── distillation_tau │ │ │ │ ├── distillation_type │ │ │ │ ├── distributed │ │ │ │ ├── drop │ │ │ │ ├── drop_path │ │ │ │ ├── epochs │ │ │ │ ├── eval │ │ │ │ ├── eval_crop_ratio │ │ │ │ ├── finetune │ │ │ │ ├── gpu │ │ │ │ ├── if_amp │ │ │ │ ├── if_continue_inf │ │ │ │ ├── if_nan2num │ │ │ │ ├── if_random_cls_token_position │ │ │ │ ├── if_random_token_rank │ │ │ │ ├── inat_category │ │ │ │ ├── input_size │ │ │ │ ├── layer_keep_ratio │ │ │ │ ├── layer_pruning_weight │ │ │ │ ├── local_rank │ │ │ │ ├── lr │ │ │ │ ├── lr_noise │ │ │ │ ├── lr_noise_pct │ │ │ │ ├── lr_noise_std │ │ │ │ ├── min_lr │ │ │ │ ├── mixup │ │ │ │ ├── mixup_mode │ │ │ │ ├── mixup_prob │ │ │ │ ├── mixup_switch_prob │ │ │ │ ├── model │ │ │ │ ├── model_ema │ │ │ │ ├── model_ema_decay │ │ │ │ ├── model_ema_force_cpu │ │ │ │ ├── model_path │ │ │ │ ├── momentum │ │ │ │ ├── n_parameters │ │ │ │ ├── num_workers │ │ │ │ ├── opt │ │ │ │ ├── opt_betas │ │ │ │ ├── opt_eps │ │ │ │ ├── output_dir │ │ │ │ ├── patience_epochs │ │ │ │ ├── pin_mem │ │ │ │ ├── rank │ │ │ │ ├── recount │ │ │ │ ├── remode │ │ │ │ ├── repeated_aug │ │ │ │ ├── reprob │ │ │ │ ├── resplit │ │ │ │ ├── resume │ │ │ │ ├── sched │ │ │ │ ├── seed │ │ │ │ ├── smoothing │ │ │ │ ├── src │ │ │ │ ├── start_epoch │ │ │ │ ├── token_keep_ratio │ │ │ │ ├── token_pruning_weight │ │ │ │ ├── train_interpolation │ │ │ │ ├── train_mode │ │ │ │ ├── unscale_lr │ │ │ │ ├── warmup_epochs │ │ │ │ ├── warmup_lr │ │ │ │ ├── weight_decay │ │ │ │ └── world_size │ │ │ └── tags │ │ │ │ ├── mlflow.runName │ │ │ │ ├── mlflow.source.name │ │ │ │ ├── mlflow.source.type │ │ │ │ └── mlflow.user │ │ │ └── meta.yaml │ │ ├── models │ │ ├── .ipynb_checkpoints │ │ │ ├── localmamba-checkpoint.py │ │ │ └── vim-checkpoint.py │ │ ├── __pycache__ │ │ │ ├── embedding.cpython-310.pyc │ │ │ ├── localmamba.cpython-310.pyc │ │ │ ├── mamba_block.cpython-310.pyc │ │ │ ├── mambareg.cpython-310.pyc │ │ │ ├── predictor.cpython-310.pyc │ │ │ └── vim.cpython-310.pyc │ │ ├── embedding.py │ │ ├── mamba_block.py │ │ ├── mambareg.py │ │ ├── predictor.py │ │ ├── videomamba.py │ │ └── vim.py │ │ ├── rope.py │ │ ├── samplers.py │ │ ├── scripts │ │ ├── .ipynb_checkpoints │ │ │ ├── localmamba-t-checkpoint.sh │ │ │ ├── mambareg-b-checkpoint.sh │ │ │ ├── mambareg-s-checkpoint.sh │ │ │ ├── vim-b-checkpoint.sh │ │ │ ├── vim-s-checkpoint.sh │ │ │ └── vim-t-checkpoint.sh │ │ ├── eval.sh │ │ ├── mambareg-b.sh │ │ └── mambareg-s.sh │ │ ├── utils.py │ │ └── vim_requirements.txt └── Vim&VideoMamba │ ├── causal-conv1d │ ├── AUTHORS │ ├── LICENSE │ ├── README.md │ ├── causal_conv1d │ │ ├── __init__.py │ │ └── causal_conv1d_interface.py │ ├── csrc │ │ ├── causal_conv1d.cpp │ │ ├── causal_conv1d.h │ │ ├── causal_conv1d_bwd.cu │ │ ├── causal_conv1d_common.h │ │ ├── causal_conv1d_fwd.cu │ │ ├── causal_conv1d_update.cu │ │ └── static_switch.h │ ├── setup.py │ └── tests │ │ └── test_causal_conv1d.py │ ├── mamba-1p1p1 │ ├── AUTHORS │ ├── LICENSE │ ├── README.md │ ├── assets │ │ └── selection.png │ ├── benchmarks │ │ └── benchmark_generation_mamba_simple.py │ ├── csrc │ │ └── selective_scan │ │ │ ├── reverse_scan.cuh │ │ │ ├── selective_scan.cpp │ │ │ ├── selective_scan.h │ │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ │ ├── selective_scan_bwd_bf16_real.cu │ │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ │ ├── selective_scan_bwd_fp16_real.cu │ │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ │ ├── selective_scan_bwd_fp32_real.cu │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ ├── selective_scan_common.h │ │ │ ├── selective_scan_fwd_bf16.cu │ │ │ ├── selective_scan_fwd_fp16.cu │ │ │ ├── selective_scan_fwd_fp32.cu │ │ │ ├── selective_scan_fwd_kernel.cuh │ │ │ ├── static_switch.h │ │ │ └── uninitialized_copy.cuh │ ├── evals │ │ └── lm_harness_eval.py │ ├── mamba_ssm │ │ ├── __init__.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── config_mamba.py │ │ │ └── mixer_seq_simple.py │ │ ├── modules │ │ │ ├── __init__.py │ │ │ └── mamba_simple.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── selective_scan_interface.py │ │ │ └── triton │ │ │ │ ├── __init__.py │ │ │ │ ├── layernorm.py │ │ │ │ └── selective_state_update.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── generation.py │ │ │ └── hf.py │ ├── setup.py │ └── tests │ │ └── ops │ │ ├── test_selective_scan.py │ │ └── triton │ │ └── test_selective_state_update.py │ └── vim │ ├── LICENSE │ ├── augment.py │ ├── calc_flops.py │ ├── datasets.py │ ├── engine.py │ ├── hubconf.py │ ├── losses.py │ ├── main.py │ ├── models │ ├── embedding.py │ ├── mamba_block.py │ ├── predictor.py │ ├── videomamba.py │ └── vim.py │ ├── rope.py │ ├── samplers.py │ ├── scripts │ ├── eval.sh │ ├── videomamba-s.sh │ ├── videomamba-t.sh │ ├── vim-b.sh │ ├── vim-s.sh │ └── vim-t.sh │ ├── utils.py │ └── vim_requirements.txt ├── README.md └── Semantic_Segmentation ├── .gitignore ├── mamba-1p1p1 ├── .github │ └── workflows │ │ └── publish.yaml ├── .gitignore ├── .gitmodules ├── AUTHORS ├── LICENSE ├── README.md ├── assets │ └── selection.png ├── benchmarks │ └── benchmark_generation_mamba_simple.py ├── csrc │ └── selective_scan │ │ ├── reverse_scan.cuh │ │ ├── selective_scan.cpp │ │ ├── selective_scan.h │ │ ├── selective_scan_bwd_bf16_complex.cu │ │ ├── selective_scan_bwd_bf16_real.cu │ │ ├── selective_scan_bwd_fp16_complex.cu │ │ ├── selective_scan_bwd_fp16_real.cu │ │ ├── selective_scan_bwd_fp32_complex.cu │ │ ├── selective_scan_bwd_fp32_real.cu │ │ ├── selective_scan_bwd_kernel.cuh │ │ ├── selective_scan_common.h │ │ ├── selective_scan_fwd_bf16.cu │ │ ├── selective_scan_fwd_fp16.cu │ │ ├── selective_scan_fwd_fp32.cu │ │ ├── selective_scan_fwd_kernel.cuh │ │ ├── static_switch.h │ │ └── uninitialized_copy.cuh ├── evals │ └── lm_harness_eval.py ├── mamba_ssm │ ├── __init__.py │ ├── models │ │ ├── __init__.py │ │ ├── config_mamba.py │ │ └── mixer_seq_simple.py │ ├── modules │ │ ├── __init__.py │ │ └── mamba_simple.py │ ├── ops │ │ ├── __init__.py │ │ ├── selective_scan_interface.py │ │ └── triton │ │ │ ├── __init__.py │ │ │ ├── layer_norm.py │ │ │ └── selective_state_update.py │ └── utils │ │ ├── __init__.py │ │ ├── generation.py │ │ └── hf.py ├── setup.py └── tests │ └── ops │ ├── test_selective_scan.py │ └── triton │ └── test_selective_state_update.py ├── readme.md ├── seg ├── README.md ├── apex.py ├── backbone │ ├── __init__.py │ └── vim.py ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k.py │ │ │ ├── ade20k_1024x1024.py │ │ │ ├── ade20k_640x640.py │ │ │ └── ade20k_ms_eval.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── plainseg_vim.py │ │ │ ├── upernet_beit.py │ │ │ └── upernet_vim.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_60k.py │ │ │ └── schedule_80k.py │ └── vim │ │ └── upernet │ │ ├── upernet_vim_small_24_512_slide_160k.py │ │ ├── upernet_vim_small_24_512_slide_60k.py │ │ ├── upernet_vim_tiny_24_512_slide_160k.py │ │ └── upernet_vim_tiny_24_512_slide_60k.py ├── mmcv_custom │ ├── __init__.py │ ├── apex_runner │ │ ├── __init__.py │ │ ├── apex_iter_based_runner.py │ │ ├── checkpoint.py │ │ └── optimizer.py │ ├── checkpoint.py │ ├── layer_decay_optimizer_constructor.py │ ├── resize_transform.py │ └── train_api.py ├── mmsegmentation │ ├── .circleci │ │ └── config.yml │ ├── .dev │ │ ├── batch_test_list.py │ │ ├── batch_train_list.txt │ │ ├── benchmark_evaluation.sh │ │ ├── benchmark_inference.py │ │ ├── benchmark_train.sh │ │ ├── check_urls.py │ │ ├── gather_benchmark_evaluation_results.py │ │ ├── gather_benchmark_train_results.py │ │ ├── gather_models.py │ │ ├── generate_benchmark_evaluation_script.py │ │ ├── generate_benchmark_train_script.py │ │ ├── log_collector │ │ │ ├── example_config.py │ │ │ ├── log_collector.py │ │ │ ├── readme.md │ │ │ └── utils.py │ │ ├── md2yml.py │ │ └── upload_modelzoo.py │ ├── .github │ │ ├── CODE_OF_CONDUCT.md │ │ ├── CONTRIBUTING.md │ │ ├── ISSUE_TEMPLATE │ │ │ ├── config.yml │ │ │ ├── error-report.md │ │ │ ├── feature_request.md │ │ │ ├── general_questions.md │ │ │ └── reimplementation_questions.md │ │ ├── pull_request_template.md │ │ └── workflows │ │ │ ├── build.yml │ │ │ ├── deploy.yml │ │ │ ├── lint.yml │ │ │ └── test_mim.yml │ ├── .gitignore │ ├── .owners.yml │ ├── .pre-commit-config.yaml │ ├── .readthedocs.yml │ ├── CITATION.cff │ ├── LICENSE │ ├── LICENSES.md │ ├── MANIFEST.in │ ├── README.md │ ├── README_zh-CN.md │ ├── configs │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── ade20k.py │ │ │ │ ├── ade20k_640x640.py │ │ │ │ ├── chase_db1.py │ │ │ │ ├── cityscapes.py │ │ │ │ ├── cityscapes_1024x1024.py │ │ │ │ ├── cityscapes_768x768.py │ │ │ │ ├── cityscapes_769x769.py │ │ │ │ ├── cityscapes_832x832.py │ │ │ │ ├── coco-stuff10k.py │ │ │ │ ├── coco-stuff164k.py │ │ │ │ ├── drive.py │ │ │ │ ├── hrf.py │ │ │ │ ├── isaid.py │ │ │ │ ├── loveda.py │ │ │ │ ├── pascal_context.py │ │ │ │ ├── pascal_context_59.py │ │ │ │ ├── pascal_voc12.py │ │ │ │ ├── pascal_voc12_aug.py │ │ │ │ ├── potsdam.py │ │ │ │ ├── stare.py │ │ │ │ └── vaihingen.py │ │ │ ├── default_runtime.py │ │ │ ├── models │ │ │ │ ├── ann_r50-d8.py │ │ │ │ ├── apcnet_r50-d8.py │ │ │ │ ├── bisenetv1_r18-d32.py │ │ │ │ ├── bisenetv2.py │ │ │ │ ├── ccnet_r50-d8.py │ │ │ │ ├── cgnet.py │ │ │ │ ├── danet_r50-d8.py │ │ │ │ ├── deeplabv3_r50-d8.py │ │ │ │ ├── deeplabv3_unet_s5-d16.py │ │ │ │ ├── deeplabv3plus_r50-d8.py │ │ │ │ ├── dmnet_r50-d8.py │ │ │ │ ├── dnl_r50-d8.py │ │ │ │ ├── dpt_vit-b16.py │ │ │ │ ├── emanet_r50-d8.py │ │ │ │ ├── encnet_r50-d8.py │ │ │ │ ├── erfnet_fcn.py │ │ │ │ ├── fast_scnn.py │ │ │ │ ├── fastfcn_r50-d32_jpu_psp.py │ │ │ │ ├── fcn_hr18.py │ │ │ │ ├── fcn_r50-d8.py │ │ │ │ ├── fcn_unet_s5-d16.py │ │ │ │ ├── fpn_poolformer_s12.py │ │ │ │ ├── fpn_r50.py │ │ │ │ ├── gcnet_r50-d8.py │ │ │ │ ├── icnet_r50-d8.py │ │ │ │ ├── isanet_r50-d8.py │ │ │ │ ├── lraspp_m-v3-d8.py │ │ │ │ ├── nonlocal_r50-d8.py │ │ │ │ ├── ocrnet_hr18.py │ │ │ │ ├── ocrnet_r50-d8.py │ │ │ │ ├── pointrend_r50.py │ │ │ │ ├── psanet_r50-d8.py │ │ │ │ ├── pspnet_r50-d8.py │ │ │ │ ├── pspnet_unet_s5-d16.py │ │ │ │ ├── segformer_mit-b0.py │ │ │ │ ├── segmenter_vit-b16_mask.py │ │ │ │ ├── setr_mla.py │ │ │ │ ├── setr_naive.py │ │ │ │ ├── setr_pup.py │ │ │ │ ├── stdc.py │ │ │ │ ├── twins_pcpvt-s_fpn.py │ │ │ │ ├── twins_pcpvt-s_upernet.py │ │ │ │ ├── upernet_beit.py │ │ │ │ ├── upernet_convnext.py │ │ │ │ ├── upernet_mae.py │ │ │ │ ├── upernet_r50.py │ │ │ │ ├── upernet_swin.py │ │ │ │ └── upernet_vit-b16_ln_mln.py │ │ │ └── schedules │ │ │ │ ├── schedule_160k.py │ │ │ │ ├── schedule_20k.py │ │ │ │ ├── schedule_320k.py │ │ │ │ ├── schedule_40k.py │ │ │ │ └── schedule_80k.py │ │ ├── ann │ │ │ ├── README.md │ │ │ ├── ann.yml │ │ │ ├── ann_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── ann_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── ann_r101-d8_512x512_160k_ade20k.py │ │ │ ├── ann_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── ann_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── ann_r101-d8_512x512_80k_ade20k.py │ │ │ ├── ann_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── ann_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── ann_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── ann_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── ann_r50-d8_512x512_160k_ade20k.py │ │ │ ├── ann_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── ann_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── ann_r50-d8_512x512_80k_ade20k.py │ │ │ ├── ann_r50-d8_769x769_40k_cityscapes.py │ │ │ └── ann_r50-d8_769x769_80k_cityscapes.py │ │ ├── apcnet │ │ │ ├── README.md │ │ │ ├── apcnet.yml │ │ │ ├── apcnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── apcnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── apcnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── apcnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── apcnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── apcnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── apcnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── apcnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── apcnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── apcnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── apcnet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── apcnet_r50-d8_769x769_80k_cityscapes.py │ │ ├── beit │ │ │ ├── README.md │ │ │ ├── beit.yml │ │ │ ├── upernet_beit-base_640x640_160k_ade20k_ms.py │ │ │ ├── upernet_beit-base_8x2_640x640_160k_ade20k.py │ │ │ ├── upernet_beit-large_fp16_640x640_160k_ade20k_ms.py │ │ │ └── upernet_beit-large_fp16_8x1_640x640_160k_ade20k.py │ │ ├── bisenetv1 │ │ │ ├── README.md │ │ │ ├── bisenetv1.yml │ │ │ ├── bisenetv1_r101-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ ├── bisenetv1_r101-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ ├── bisenetv1_r18-d32_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r18-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r18-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ ├── bisenetv1_r18-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ ├── bisenetv1_r50-d32_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r50-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r50-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ └── bisenetv1_r50-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ ├── bisenetv2 │ │ │ ├── README.md │ │ │ ├── bisenetv2.yml │ │ │ ├── bisenetv2_fcn_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv2_fcn_fp16_4x4_1024x1024_160k_cityscapes.py │ │ │ └── bisenetv2_fcn_ohem_4x4_1024x1024_160k_cityscapes.py │ │ ├── ccnet │ │ │ ├── README.md │ │ │ ├── ccnet.yml │ │ │ ├── ccnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── ccnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── ccnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── ccnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── ccnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── ccnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── ccnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── ccnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── ccnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── ccnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── ccnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── ccnet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── ccnet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── ccnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── ccnet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── ccnet_r50-d8_769x769_80k_cityscapes.py │ │ ├── cgnet │ │ │ ├── README.md │ │ │ ├── cgnet.yml │ │ │ ├── cgnet_512x1024_60k_cityscapes.py │ │ │ └── cgnet_680x680_60k_cityscapes.py │ │ ├── convnext │ │ │ ├── README.md │ │ │ ├── convnext.yml │ │ │ ├── upernet_convnext_base_fp16_512x512_160k_ade20k.py │ │ │ ├── upernet_convnext_base_fp16_640x640_160k_ade20k.py │ │ │ ├── upernet_convnext_large_fp16_640x640_160k_ade20k.py │ │ │ ├── upernet_convnext_small_fp16_512x512_160k_ade20k.py │ │ │ ├── upernet_convnext_tiny_fp16_512x512_160k_ade20k.py │ │ │ └── upernet_convnext_xlarge_fp16_640x640_160k_ade20k.py │ │ ├── danet │ │ │ ├── README.md │ │ │ ├── danet.yml │ │ │ ├── danet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── danet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── danet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── danet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── danet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── danet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── danet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── danet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── danet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── danet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── danet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── danet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── danet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── danet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── danet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── danet_r50-d8_769x769_80k_cityscapes.py │ │ ├── deeplabv3 │ │ │ ├── README.md │ │ │ ├── deeplabv3.yml │ │ │ ├── deeplabv3_r101-d16-mg124_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3_r101-d16-mg124_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── deeplabv3_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── deeplabv3_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_160k_coco-stuff164k.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_20k_coco-stuff10k.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_320k_coco-stuff164k.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_40k_coco-stuff10k.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_80k_coco-stuff164k.py │ │ │ ├── deeplabv3_r101-d8_512x512_80k_ade20k.py │ │ │ ├── deeplabv3_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r18-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r18-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r18b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r18b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3_r50-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3_r50-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3_r50-d8_480x480_80k_pascal_context_59.py │ │ │ ├── deeplabv3_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── deeplabv3_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_160k_coco-stuff164k.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_20k_coco-stuff10k.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_320k_coco-stuff164k.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_40k_coco-stuff10k.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_80k_coco-stuff164k.py │ │ │ ├── deeplabv3_r50-d8_512x512_80k_ade20k.py │ │ │ ├── deeplabv3_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r50b-d8_512x1024_80k_cityscapes.py │ │ │ └── deeplabv3_r50b-d8_769x769_80k_cityscapes.py │ │ ├── deeplabv3plus │ │ │ ├── README.md │ │ │ ├── deeplabv3plus.yml │ │ │ ├── deeplabv3plus_r101-d16-mg124_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d16-mg124_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r101-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── deeplabv3plus_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_80k_ade20k.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_80k_loveda.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_80k_potsdam.py │ │ │ ├── deeplabv3plus_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r18-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── deeplabv3plus_r18-d8_4x4_896x896_80k_isaid.py │ │ │ ├── deeplabv3plus_r18-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r18-d8_512x512_80k_loveda.py │ │ │ ├── deeplabv3plus_r18-d8_512x512_80k_potsdam.py │ │ │ ├── deeplabv3plus_r18-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r18b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r18b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3plus_r50-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r50-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3plus_r50-d8_480x480_80k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r50-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── deeplabv3plus_r50-d8_4x4_896x896_80k_isaid.py │ │ │ ├── deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_80k_ade20k.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_80k_loveda.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_80k_potsdam.py │ │ │ ├── deeplabv3plus_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r50b-d8_512x1024_80k_cityscapes.py │ │ │ └── deeplabv3plus_r50b-d8_769x769_80k_cityscapes.py │ │ ├── dmnet │ │ │ ├── README.md │ │ │ ├── dmnet.yml │ │ │ ├── dmnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── dmnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── dmnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── dmnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── dmnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── dmnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── dmnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── dmnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── dmnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── dmnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── dmnet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── dmnet_r50-d8_769x769_80k_cityscapes.py │ │ ├── dnlnet │ │ │ ├── README.md │ │ │ ├── dnl_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── dnl_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── dnl_r101-d8_512x512_160k_ade20k.py │ │ │ ├── dnl_r101-d8_512x512_80k_ade20k.py │ │ │ ├── dnl_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── dnl_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── dnl_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── dnl_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── dnl_r50-d8_512x512_160k_ade20k.py │ │ │ ├── dnl_r50-d8_512x512_80k_ade20k.py │ │ │ ├── dnl_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── dnl_r50-d8_769x769_80k_cityscapes.py │ │ │ └── dnlnet.yml │ │ ├── dpt │ │ │ ├── README.md │ │ │ ├── dpt.yml │ │ │ └── dpt_vit-b16_512x512_160k_ade20k.py │ │ ├── emanet │ │ │ ├── README.md │ │ │ ├── emanet.yml │ │ │ ├── emanet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── emanet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── emanet_r50-d8_512x1024_80k_cityscapes.py │ │ │ └── emanet_r50-d8_769x769_80k_cityscapes.py │ │ ├── encnet │ │ │ ├── README.md │ │ │ ├── encnet.yml │ │ │ ├── encnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── encnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── encnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── encnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── encnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── encnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── encnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── encnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── encnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── encnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── encnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── encnet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── encnet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── encnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── encnet_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── encnet_r50-d8_769x769_80k_cityscapes.py │ │ │ └── encnet_r50s-d8_512x512_80k_ade20k.py │ │ ├── erfnet │ │ │ ├── README.md │ │ │ ├── erfnet.yml │ │ │ └── erfnet_fcn_4x4_512x1024_160k_cityscapes.py │ │ ├── fastfcn │ │ │ ├── README.md │ │ │ ├── fastfcn.yml │ │ │ ├── fastfcn_r50-d32_jpu_aspp_4x4_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_aspp_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_aspp_512x512_160k_ade20k.py │ │ │ ├── fastfcn_r50-d32_jpu_aspp_512x512_80k_ade20k.py │ │ │ ├── fastfcn_r50-d32_jpu_enc_4x4_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_enc_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_enc_512x512_160k_ade20k.py │ │ │ ├── fastfcn_r50-d32_jpu_enc_512x512_80k_ade20k.py │ │ │ ├── fastfcn_r50-d32_jpu_psp_4x4_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_psp_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_psp_512x512_160k_ade20k.py │ │ │ └── fastfcn_r50-d32_jpu_psp_512x512_80k_ade20k.py │ │ ├── fastscnn │ │ │ ├── README.md │ │ │ ├── fast_scnn_lr0.12_8x4_160k_cityscapes.py │ │ │ └── fastscnn.yml │ │ ├── fcn │ │ │ ├── README.md │ │ │ ├── fcn.yml │ │ │ ├── fcn_d6_r101-d16_512x1024_40k_cityscapes.py │ │ │ ├── fcn_d6_r101-d16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r101-d16_769x769_40k_cityscapes.py │ │ │ ├── fcn_d6_r101-d16_769x769_80k_cityscapes.py │ │ │ ├── fcn_d6_r101b-d16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r101b-d16_769x769_80k_cityscapes.py │ │ │ ├── fcn_d6_r50-d16_512x1024_40k_cityscapes.py │ │ │ ├── fcn_d6_r50-d16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r50-d16_769x769_40k_cityscapes.py │ │ │ ├── fcn_d6_r50-d16_769x769_80k_cityscapes.py │ │ │ ├── fcn_d6_r50b-d16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r50b-d16_769x769_80k_cityscapes.py │ │ │ ├── fcn_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── fcn_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── fcn_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── fcn_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r101-d8_512x512_160k_ade20k.py │ │ │ ├── fcn_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── fcn_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── fcn_r101-d8_512x512_80k_ade20k.py │ │ │ ├── fcn_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── fcn_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r18-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r18-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r18b-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r18b-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r50-d8_480x480_40k_pascal_context.py │ │ │ ├── fcn_r50-d8_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_r50-d8_480x480_80k_pascal_context.py │ │ │ ├── fcn_r50-d8_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── fcn_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r50-d8_512x512_160k_ade20k.py │ │ │ ├── fcn_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── fcn_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── fcn_r50-d8_512x512_80k_ade20k.py │ │ │ ├── fcn_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── fcn_r50-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r50b-d8_512x1024_80k_cityscapes.py │ │ │ └── fcn_r50b-d8_769x769_80k_cityscapes.py │ │ ├── gcnet │ │ │ ├── README.md │ │ │ ├── gcnet.yml │ │ │ ├── gcnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── gcnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── gcnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── gcnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── gcnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── gcnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── gcnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── gcnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── gcnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── gcnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── gcnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── gcnet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── gcnet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── gcnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── gcnet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── gcnet_r50-d8_769x769_80k_cityscapes.py │ │ ├── hrnet │ │ │ ├── README.md │ │ │ ├── fcn_hr18_480x480_40k_pascal_context.py │ │ │ ├── fcn_hr18_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_hr18_480x480_80k_pascal_context.py │ │ │ ├── fcn_hr18_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_hr18_4x4_512x512_80k_vaihingen.py │ │ │ ├── fcn_hr18_4x4_896x896_80k_isaid.py │ │ │ ├── fcn_hr18_512x1024_160k_cityscapes.py │ │ │ ├── fcn_hr18_512x1024_40k_cityscapes.py │ │ │ ├── fcn_hr18_512x1024_80k_cityscapes.py │ │ │ ├── fcn_hr18_512x512_160k_ade20k.py │ │ │ ├── fcn_hr18_512x512_20k_voc12aug.py │ │ │ ├── fcn_hr18_512x512_40k_voc12aug.py │ │ │ ├── fcn_hr18_512x512_80k_ade20k.py │ │ │ ├── fcn_hr18_512x512_80k_loveda.py │ │ │ ├── fcn_hr18_512x512_80k_potsdam.py │ │ │ ├── fcn_hr18s_480x480_40k_pascal_context.py │ │ │ ├── fcn_hr18s_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_hr18s_480x480_80k_pascal_context.py │ │ │ ├── fcn_hr18s_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_hr18s_4x4_512x512_80k_vaihingen.py │ │ │ ├── fcn_hr18s_4x4_896x896_80k_isaid.py │ │ │ ├── fcn_hr18s_512x1024_160k_cityscapes.py │ │ │ ├── fcn_hr18s_512x1024_40k_cityscapes.py │ │ │ ├── fcn_hr18s_512x1024_80k_cityscapes.py │ │ │ ├── fcn_hr18s_512x512_160k_ade20k.py │ │ │ ├── fcn_hr18s_512x512_20k_voc12aug.py │ │ │ ├── fcn_hr18s_512x512_40k_voc12aug.py │ │ │ ├── fcn_hr18s_512x512_80k_ade20k.py │ │ │ ├── fcn_hr18s_512x512_80k_loveda.py │ │ │ ├── fcn_hr18s_512x512_80k_potsdam.py │ │ │ ├── fcn_hr48_480x480_40k_pascal_context.py │ │ │ ├── fcn_hr48_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_hr48_480x480_80k_pascal_context.py │ │ │ ├── fcn_hr48_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_hr48_4x4_512x512_80k_vaihingen.py │ │ │ ├── fcn_hr48_4x4_896x896_80k_isaid.py │ │ │ ├── fcn_hr48_512x1024_160k_cityscapes.py │ │ │ ├── fcn_hr48_512x1024_40k_cityscapes.py │ │ │ ├── fcn_hr48_512x1024_80k_cityscapes.py │ │ │ ├── fcn_hr48_512x512_160k_ade20k.py │ │ │ ├── fcn_hr48_512x512_20k_voc12aug.py │ │ │ ├── fcn_hr48_512x512_40k_voc12aug.py │ │ │ ├── fcn_hr48_512x512_80k_ade20k.py │ │ │ ├── fcn_hr48_512x512_80k_loveda.py │ │ │ ├── fcn_hr48_512x512_80k_potsdam.py │ │ │ └── hrnet.yml │ │ ├── icnet │ │ │ ├── README.md │ │ │ ├── icnet.yml │ │ │ ├── icnet_r101-d8_832x832_160k_cityscapes.py │ │ │ ├── icnet_r101-d8_832x832_80k_cityscapes.py │ │ │ ├── icnet_r101-d8_in1k-pre_832x832_160k_cityscapes.py │ │ │ ├── icnet_r101-d8_in1k-pre_832x832_80k_cityscapes.py │ │ │ ├── icnet_r18-d8_832x832_160k_cityscapes.py │ │ │ ├── icnet_r18-d8_832x832_80k_cityscapes.py │ │ │ ├── icnet_r18-d8_in1k-pre_832x832_160k_cityscapes.py │ │ │ ├── icnet_r18-d8_in1k-pre_832x832_80k_cityscapes.py │ │ │ ├── icnet_r50-d8_832x832_160k_cityscapes.py │ │ │ ├── icnet_r50-d8_832x832_80k_cityscapes.py │ │ │ ├── icnet_r50-d8_in1k-pre_832x832_160k_cityscapes.py │ │ │ └── icnet_r50-d8_in1k-pre_832x832_80k_cityscapes.py │ │ ├── isanet │ │ │ ├── README.md │ │ │ ├── isanet.yml │ │ │ ├── isanet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── isanet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── isanet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── isanet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── isanet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── isanet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── isanet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── isanet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── isanet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── isanet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── isanet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── isanet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── isanet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── isanet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── isanet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── isanet_r50-d8_769x769_80k_cityscapes.py │ │ ├── knet │ │ │ ├── README.md │ │ │ ├── knet.yml │ │ │ ├── knet_s3_deeplabv3_r50-d8_8x2_512x512_adamw_80k_ade20k.py │ │ │ ├── knet_s3_fcn_r50-d8_8x2_512x512_adamw_80k_ade20k.py │ │ │ ├── knet_s3_pspnet_r50-d8_8x2_512x512_adamw_80k_ade20k.py │ │ │ ├── knet_s3_upernet_r50-d8_8x2_512x512_adamw_80k_ade20k.py │ │ │ ├── knet_s3_upernet_swin-l_8x2_512x512_adamw_80k_ade20k.py │ │ │ ├── knet_s3_upernet_swin-l_8x2_640x640_adamw_80k_ade20k.py │ │ │ └── knet_s3_upernet_swin-t_8x2_512x512_adamw_80k_ade20k.py │ │ ├── mae │ │ │ ├── README.md │ │ │ ├── mae.yml │ │ │ ├── upernet_mae-base_fp16_512x512_160k_ade20k_ms.py │ │ │ └── upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py │ │ ├── mobilenet_v2 │ │ │ ├── README.md │ │ │ ├── deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_m-v2-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py │ │ │ ├── fcn_m-v2-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_m-v2-d8_512x512_160k_ade20k.py │ │ │ ├── mobilenet_v2.yml │ │ │ ├── pspnet_m-v2-d8_512x1024_80k_cityscapes.py │ │ │ └── pspnet_m-v2-d8_512x512_160k_ade20k.py │ │ ├── mobilenet_v3 │ │ │ ├── README.md │ │ │ ├── lraspp_m-v3-d8_512x1024_320k_cityscapes.py │ │ │ ├── lraspp_m-v3-d8_scratch_512x1024_320k_cityscapes.py │ │ │ ├── lraspp_m-v3s-d8_512x1024_320k_cityscapes.py │ │ │ ├── lraspp_m-v3s-d8_scratch_512x1024_320k_cityscapes.py │ │ │ └── mobilenet_v3.yml │ │ ├── nonlocal_net │ │ │ ├── README.md │ │ │ ├── nonlocal_net.yml │ │ │ ├── nonlocal_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── nonlocal_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── nonlocal_r101-d8_512x512_160k_ade20k.py │ │ │ ├── nonlocal_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── nonlocal_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── nonlocal_r101-d8_512x512_80k_ade20k.py │ │ │ ├── nonlocal_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── nonlocal_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── nonlocal_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── nonlocal_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── nonlocal_r50-d8_512x512_160k_ade20k.py │ │ │ ├── nonlocal_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── nonlocal_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── nonlocal_r50-d8_512x512_80k_ade20k.py │ │ │ ├── nonlocal_r50-d8_769x769_40k_cityscapes.py │ │ │ └── nonlocal_r50-d8_769x769_80k_cityscapes.py │ │ ├── ocrnet │ │ │ ├── README.md │ │ │ ├── ocrnet.yml │ │ │ ├── ocrnet_hr18_512x1024_160k_cityscapes.py │ │ │ ├── ocrnet_hr18_512x1024_40k_cityscapes.py │ │ │ ├── ocrnet_hr18_512x1024_80k_cityscapes.py │ │ │ ├── ocrnet_hr18_512x512_160k_ade20k.py │ │ │ ├── ocrnet_hr18_512x512_20k_voc12aug.py │ │ │ ├── ocrnet_hr18_512x512_40k_voc12aug.py │ │ │ ├── ocrnet_hr18_512x512_80k_ade20k.py │ │ │ ├── ocrnet_hr18s_512x1024_160k_cityscapes.py │ │ │ ├── ocrnet_hr18s_512x1024_40k_cityscapes.py │ │ │ ├── ocrnet_hr18s_512x1024_80k_cityscapes.py │ │ │ ├── ocrnet_hr18s_512x512_160k_ade20k.py │ │ │ ├── ocrnet_hr18s_512x512_20k_voc12aug.py │ │ │ ├── ocrnet_hr18s_512x512_40k_voc12aug.py │ │ │ ├── ocrnet_hr18s_512x512_80k_ade20k.py │ │ │ ├── ocrnet_hr48_512x1024_160k_cityscapes.py │ │ │ ├── ocrnet_hr48_512x1024_40k_cityscapes.py │ │ │ ├── ocrnet_hr48_512x1024_80k_cityscapes.py │ │ │ ├── ocrnet_hr48_512x512_160k_ade20k.py │ │ │ ├── ocrnet_hr48_512x512_20k_voc12aug.py │ │ │ ├── ocrnet_hr48_512x512_40k_voc12aug.py │ │ │ ├── ocrnet_hr48_512x512_80k_ade20k.py │ │ │ ├── ocrnet_r101-d8_512x1024_40k_b16_cityscapes.py │ │ │ ├── ocrnet_r101-d8_512x1024_40k_b8_cityscapes.py │ │ │ └── ocrnet_r101-d8_512x1024_80k_b16_cityscapes.py │ │ ├── point_rend │ │ │ ├── README.md │ │ │ ├── point_rend.yml │ │ │ ├── pointrend_r101_512x1024_80k_cityscapes.py │ │ │ ├── pointrend_r101_512x512_160k_ade20k.py │ │ │ ├── pointrend_r50_512x1024_80k_cityscapes.py │ │ │ └── pointrend_r50_512x512_160k_ade20k.py │ │ ├── poolformer │ │ │ ├── README.md │ │ │ ├── fpn_poolformer_m36_8x4_512x512_40k_ade20k.py │ │ │ ├── fpn_poolformer_m48_8x4_512x512_40k_ade20k.py │ │ │ ├── fpn_poolformer_s12_8x4_512x512_40k_ade20k.py │ │ │ ├── fpn_poolformer_s24_8x4_512x512_40k_ade20k.py │ │ │ ├── fpn_poolformer_s36_8x4_512x512_40k_ade20k.py │ │ │ └── poolformer.yml │ │ ├── psanet │ │ │ ├── README.md │ │ │ ├── psanet.yml │ │ │ ├── psanet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── psanet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── psanet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── psanet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── psanet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── psanet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── psanet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── psanet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── psanet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── psanet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── psanet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── psanet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── psanet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── psanet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── psanet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── psanet_r50-d8_769x769_80k_cityscapes.py │ │ ├── pspnet │ │ │ ├── README.md │ │ │ ├── pspnet.yml │ │ │ ├── pspnet_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── pspnet_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── pspnet_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── pspnet_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── pspnet_r101-d8_4x4_512x512_80k_potsdam.py │ │ │ ├── pspnet_r101-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── pspnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── pspnet_r101-d8_512x1024_40k_dark.py │ │ │ ├── pspnet_r101-d8_512x1024_40k_night_driving.py │ │ │ ├── pspnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── pspnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── pspnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_160k_coco-stuff164k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_20k_coco-stuff10k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_320k_coco-stuff164k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_40k_coco-stuff10k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_80k_coco-stuff164k.py │ │ │ ├── pspnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── pspnet_r101-d8_512x512_80k_loveda.py │ │ │ ├── pspnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── pspnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r101b-d8_512x1024_80k_dark.py │ │ │ ├── pspnet_r101b-d8_512x1024_80k_night_driving.py │ │ │ ├── pspnet_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r18-d8_4x4_512x512_80k_potsdam.py │ │ │ ├── pspnet_r18-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── pspnet_r18-d8_4x4_896x896_80k_isaid.py │ │ │ ├── pspnet_r18-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r18-d8_512x512_80k_loveda.py │ │ │ ├── pspnet_r18-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r18b-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r18b-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r50-d32_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r50-d32_rsb-pretrain_512x1024_adamw_80k_cityscapes.py │ │ │ ├── pspnet_r50-d8_480x480_40k_pascal_context.py │ │ │ ├── pspnet_r50-d8_480x480_40k_pascal_context_59.py │ │ │ ├── pspnet_r50-d8_480x480_80k_pascal_context.py │ │ │ ├── pspnet_r50-d8_480x480_80k_pascal_context_59.py │ │ │ ├── pspnet_r50-d8_4x4_512x512_80k_potsdam.py │ │ │ ├── pspnet_r50-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── pspnet_r50-d8_4x4_896x896_80k_isaid.py │ │ │ ├── pspnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── pspnet_r50-d8_512x1024_40k_dark.py │ │ │ ├── pspnet_r50-d8_512x1024_40k_night_driving.py │ │ │ ├── pspnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r50-d8_512x1024_80k_dark.py │ │ │ ├── pspnet_r50-d8_512x1024_80k_night_driving.py │ │ │ ├── pspnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── pspnet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── pspnet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_160k_coco-stuff164k.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_20k_coco-stuff10k.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_320k_coco-stuff164k.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_40k_coco-stuff10k.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_80k_coco-stuff164k.py │ │ │ ├── pspnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── pspnet_r50-d8_512x512_80k_loveda.py │ │ │ ├── pspnet_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── pspnet_r50-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r50-d8_rsb-pretrain_512x1024_adamw_80k_cityscapes.py │ │ │ ├── pspnet_r50b-d32_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r50b-d8_512x1024_80k_cityscapes.py │ │ │ └── pspnet_r50b-d8_769x769_80k_cityscapes.py │ │ ├── resnest │ │ │ ├── README.md │ │ │ ├── deeplabv3_s101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_s101-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3plus_s101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_s101-d8_512x512_160k_ade20k.py │ │ │ ├── fcn_s101-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_s101-d8_512x512_160k_ade20k.py │ │ │ ├── pspnet_s101-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_s101-d8_512x512_160k_ade20k.py │ │ │ └── resnest.yml │ │ ├── segformer │ │ │ ├── README.md │ │ │ ├── segformer.yml │ │ │ ├── segformer_mit-b0_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b0_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b1_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b1_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b2_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b2_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b3_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b3_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b4_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b4_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b5_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b5_640x640_160k_ade20k.py │ │ │ └── segformer_mit-b5_8x1_1024x1024_160k_cityscapes.py │ │ ├── segmenter │ │ │ ├── README.md │ │ │ ├── segmenter.yml │ │ │ ├── segmenter_vit-b_mask_8x1_512x512_160k_ade20k.py │ │ │ ├── segmenter_vit-l_mask_8x1_640x640_160k_ade20k.py │ │ │ ├── segmenter_vit-s_linear_8x1_512x512_160k_ade20k.py │ │ │ ├── segmenter_vit-s_mask_8x1_512x512_160k_ade20k.py │ │ │ └── segmenter_vit-t_mask_8x1_512x512_160k_ade20k.py │ │ ├── sem_fpn │ │ │ ├── README.md │ │ │ ├── fpn_r101_512x1024_80k_cityscapes.py │ │ │ ├── fpn_r101_512x512_160k_ade20k.py │ │ │ ├── fpn_r50_512x1024_80k_cityscapes.py │ │ │ ├── fpn_r50_512x512_160k_ade20k.py │ │ │ └── sem_fpn.yml │ │ ├── setr │ │ │ ├── README.md │ │ │ ├── setr.yml │ │ │ ├── setr_mla_512x512_160k_b16_ade20k.py │ │ │ ├── setr_mla_512x512_160k_b8_ade20k.py │ │ │ ├── setr_naive_512x512_160k_b16_ade20k.py │ │ │ ├── setr_pup_512x512_160k_b16_ade20k.py │ │ │ ├── setr_vit-large_mla_8x1_768x768_80k_cityscapes.py │ │ │ ├── setr_vit-large_naive_8x1_768x768_80k_cityscapes.py │ │ │ └── setr_vit-large_pup_8x1_768x768_80k_cityscapes.py │ │ ├── stdc │ │ │ ├── README.md │ │ │ ├── stdc.yml │ │ │ ├── stdc1_512x1024_80k_cityscapes.py │ │ │ ├── stdc1_in1k-pre_512x1024_80k_cityscapes.py │ │ │ ├── stdc2_512x1024_80k_cityscapes.py │ │ │ └── stdc2_in1k-pre_512x1024_80k_cityscapes.py │ │ ├── swin │ │ │ ├── README.md │ │ │ ├── swin.yml │ │ │ ├── upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_1K.py │ │ │ ├── upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_22K.py │ │ │ ├── upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py │ │ │ ├── upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_22K.py │ │ │ ├── upernet_swin_large_patch4_window12_512x512_pretrain_384x384_22K_160k_ade20k.py │ │ │ ├── upernet_swin_large_patch4_window7_512x512_pretrain_224x224_22K_160k_ade20k.py │ │ │ ├── upernet_swin_small_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py │ │ │ └── upernet_swin_tiny_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py │ │ ├── twins │ │ │ ├── README.md │ │ │ ├── twins.yml │ │ │ ├── twins_pcpvt-b_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_pcpvt-b_uperhead_8x2_512x512_160k_ade20k.py │ │ │ ├── twins_pcpvt-l_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_pcpvt-l_uperhead_8x2_512x512_160k_ade20k.py │ │ │ ├── twins_pcpvt-s_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_pcpvt-s_uperhead_8x4_512x512_160k_ade20k.py │ │ │ ├── twins_svt-b_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_svt-b_uperhead_8x2_512x512_160k_ade20k.py │ │ │ ├── twins_svt-l_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_svt-l_uperhead_8x2_512x512_160k_ade20k.py │ │ │ ├── twins_svt-s_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ └── twins_svt-s_uperhead_8x2_512x512_160k_ade20k.py │ │ ├── unet │ │ │ ├── README.md │ │ │ ├── deeplabv3_unet_s5-d16_128x128_40k_chase_db1.py │ │ │ ├── deeplabv3_unet_s5-d16_128x128_40k_stare.py │ │ │ ├── deeplabv3_unet_s5-d16_256x256_40k_hrf.py │ │ │ ├── deeplabv3_unet_s5-d16_64x64_40k_drive.py │ │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py │ │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py │ │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py │ │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_64x64_40k_drive.py │ │ │ ├── fcn_unet_s5-d16_128x128_40k_chase_db1.py │ │ │ ├── fcn_unet_s5-d16_128x128_40k_stare.py │ │ │ ├── fcn_unet_s5-d16_256x256_40k_hrf.py │ │ │ ├── fcn_unet_s5-d16_4x4_512x1024_160k_cityscapes.py │ │ │ ├── fcn_unet_s5-d16_64x64_40k_drive.py │ │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py │ │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py │ │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py │ │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_64x64_40k_drive.py │ │ │ ├── pspnet_unet_s5-d16_128x128_40k_chase_db1.py │ │ │ ├── pspnet_unet_s5-d16_128x128_40k_stare.py │ │ │ ├── pspnet_unet_s5-d16_256x256_40k_hrf.py │ │ │ ├── pspnet_unet_s5-d16_64x64_40k_drive.py │ │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py │ │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py │ │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py │ │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_64x64_40k_drive.py │ │ │ └── unet.yml │ │ ├── upernet │ │ │ ├── README.md │ │ │ ├── upernet.yml │ │ │ ├── upernet_r101_512x1024_40k_cityscapes.py │ │ │ ├── upernet_r101_512x1024_80k_cityscapes.py │ │ │ ├── upernet_r101_512x512_160k_ade20k.py │ │ │ ├── upernet_r101_512x512_20k_voc12aug.py │ │ │ ├── upernet_r101_512x512_40k_voc12aug.py │ │ │ ├── upernet_r101_512x512_80k_ade20k.py │ │ │ ├── upernet_r101_769x769_40k_cityscapes.py │ │ │ ├── upernet_r101_769x769_80k_cityscapes.py │ │ │ ├── upernet_r18_512x1024_40k_cityscapes.py │ │ │ ├── upernet_r18_512x1024_80k_cityscapes.py │ │ │ ├── upernet_r18_512x512_160k_ade20k.py │ │ │ ├── upernet_r18_512x512_20k_voc12aug.py │ │ │ ├── upernet_r18_512x512_40k_voc12aug.py │ │ │ ├── upernet_r18_512x512_80k_ade20k.py │ │ │ ├── upernet_r50_512x1024_40k_cityscapes.py │ │ │ ├── upernet_r50_512x1024_80k_cityscapes.py │ │ │ ├── upernet_r50_512x512_160k_ade20k.py │ │ │ ├── upernet_r50_512x512_20k_voc12aug.py │ │ │ ├── upernet_r50_512x512_40k_voc12aug.py │ │ │ ├── upernet_r50_512x512_80k_ade20k.py │ │ │ ├── upernet_r50_769x769_40k_cityscapes.py │ │ │ └── upernet_r50_769x769_80k_cityscapes.py │ │ └── vit │ │ │ ├── README.md │ │ │ ├── upernet_deit-b16_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-b16_512x512_80k_ade20k.py │ │ │ ├── upernet_deit-b16_ln_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-b16_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-s16_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-s16_512x512_80k_ade20k.py │ │ │ ├── upernet_deit-s16_ln_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-s16_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_vit-b16_ln_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_vit-b16_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_vit-b16_mln_512x512_80k_ade20k.py │ │ │ └── vit.yml │ ├── demo │ │ ├── MMSegmentation_Tutorial.ipynb │ │ ├── demo.png │ │ ├── image_demo.py │ │ ├── inference_demo.ipynb │ │ └── video_demo.py │ ├── docker │ │ ├── Dockerfile │ │ └── serve │ │ │ ├── Dockerfile │ │ │ ├── config.properties │ │ │ └── entrypoint.sh │ ├── docs │ │ ├── en │ │ │ ├── Makefile │ │ │ ├── _static │ │ │ │ ├── css │ │ │ │ │ └── readthedocs.css │ │ │ │ └── images │ │ │ │ │ └── mmsegmentation.png │ │ │ ├── api.rst │ │ │ ├── changelog.md │ │ │ ├── conf.py │ │ │ ├── dataset_prepare.md │ │ │ ├── faq.md │ │ │ ├── get_started.md │ │ │ ├── index.rst │ │ │ ├── inference.md │ │ │ ├── make.bat │ │ │ ├── model_zoo.md │ │ │ ├── stat.py │ │ │ ├── switch_language.md │ │ │ ├── train.md │ │ │ ├── tutorials │ │ │ │ ├── config.md │ │ │ │ ├── customize_datasets.md │ │ │ │ ├── customize_models.md │ │ │ │ ├── customize_runtime.md │ │ │ │ ├── data_pipeline.md │ │ │ │ ├── index.rst │ │ │ │ └── training_tricks.md │ │ │ └── useful_tools.md │ │ └── zh_cn │ │ │ ├── Makefile │ │ │ ├── _static │ │ │ ├── css │ │ │ │ └── readthedocs.css │ │ │ └── images │ │ │ │ └── mmsegmentation.png │ │ │ ├── api.rst │ │ │ ├── conf.py │ │ │ ├── dataset_prepare.md │ │ │ ├── faq.md │ │ │ ├── get_started.md │ │ │ ├── imgs │ │ │ ├── qq_group_qrcode.jpg │ │ │ └── zhihu_qrcode.jpg │ │ │ ├── index.rst │ │ │ ├── inference.md │ │ │ ├── make.bat │ │ │ ├── model_zoo.md │ │ │ ├── stat.py │ │ │ ├── switch_language.md │ │ │ ├── train.md │ │ │ ├── tutorials │ │ │ ├── config.md │ │ │ ├── customize_datasets.md │ │ │ ├── customize_models.md │ │ │ ├── customize_runtime.md │ │ │ ├── data_pipeline.md │ │ │ ├── index.rst │ │ │ └── training_tricks.md │ │ │ └── useful_tools.md │ ├── mmseg │ │ ├── __init__.py │ │ ├── apis │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── test.py │ │ │ └── train.py │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── evaluation │ │ │ │ ├── __init__.py │ │ │ │ ├── class_names.py │ │ │ │ ├── eval_hooks.py │ │ │ │ └── metrics.py │ │ │ ├── hook │ │ │ │ ├── __init__.py │ │ │ │ └── wandblogger_hook.py │ │ │ ├── optimizers │ │ │ │ ├── __init__.py │ │ │ │ └── layer_decay_optimizer_constructor.py │ │ │ ├── seg │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ └── sampler │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_pixel_sampler.py │ │ │ │ │ └── ohem_pixel_sampler.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── dist_util.py │ │ │ │ └── misc.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── ade.py │ │ │ ├── builder.py │ │ │ ├── chase_db1.py │ │ │ ├── cityscapes.py │ │ │ ├── coco_stuff.py │ │ │ ├── custom.py │ │ │ ├── dark_zurich.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── drive.py │ │ │ ├── hrf.py │ │ │ ├── isaid.py │ │ │ ├── isprs.py │ │ │ ├── loveda.py │ │ │ ├── night_driving.py │ │ │ ├── pascal_context.py │ │ │ ├── pipelines │ │ │ │ ├── __init__.py │ │ │ │ ├── compose.py │ │ │ │ ├── formating.py │ │ │ │ ├── formatting.py │ │ │ │ ├── loading.py │ │ │ │ ├── test_time_aug.py │ │ │ │ └── transforms.py │ │ │ ├── potsdam.py │ │ │ ├── samplers │ │ │ │ ├── __init__.py │ │ │ │ └── distributed_sampler.py │ │ │ ├── stare.py │ │ │ └── voc.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── beit.py │ │ │ │ ├── bisenetv1.py │ │ │ │ ├── bisenetv2.py │ │ │ │ ├── cgnet.py │ │ │ │ ├── erfnet.py │ │ │ │ ├── fast_scnn.py │ │ │ │ ├── hrnet.py │ │ │ │ ├── icnet.py │ │ │ │ ├── mae.py │ │ │ │ ├── mit.py │ │ │ │ ├── mobilenet_v2.py │ │ │ │ ├── mobilenet_v3.py │ │ │ │ ├── resnest.py │ │ │ │ ├── resnet.py │ │ │ │ ├── resnext.py │ │ │ │ ├── stdc.py │ │ │ │ ├── swin.py │ │ │ │ ├── timm_backbone.py │ │ │ │ ├── twins.py │ │ │ │ ├── unet.py │ │ │ │ └── vit.py │ │ │ ├── builder.py │ │ │ ├── decode_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── ann_head.py │ │ │ │ ├── apc_head.py │ │ │ │ ├── aspp_head.py │ │ │ │ ├── cascade_decode_head.py │ │ │ │ ├── cc_head.py │ │ │ │ ├── da_head.py │ │ │ │ ├── decode_head.py │ │ │ │ ├── dm_head.py │ │ │ │ ├── dnl_head.py │ │ │ │ ├── dpt_head.py │ │ │ │ ├── ema_head.py │ │ │ │ ├── enc_head.py │ │ │ │ ├── fcn_head.py │ │ │ │ ├── fpn_head.py │ │ │ │ ├── gc_head.py │ │ │ │ ├── isa_head.py │ │ │ │ ├── knet_head.py │ │ │ │ ├── lraspp_head.py │ │ │ │ ├── nl_head.py │ │ │ │ ├── ocr_head.py │ │ │ │ ├── point_head.py │ │ │ │ ├── psa_head.py │ │ │ │ ├── psp_head.py │ │ │ │ ├── segformer_head.py │ │ │ │ ├── segmenter_mask_head.py │ │ │ │ ├── sep_aspp_head.py │ │ │ │ ├── sep_fcn_head.py │ │ │ │ ├── setr_mla_head.py │ │ │ │ ├── setr_up_head.py │ │ │ │ ├── stdc_head.py │ │ │ │ └── uper_head.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── accuracy.py │ │ │ │ ├── cross_entropy_loss.py │ │ │ │ ├── dice_loss.py │ │ │ │ ├── focal_loss.py │ │ │ │ ├── lovasz_loss.py │ │ │ │ ├── tversky_loss.py │ │ │ │ └── utils.py │ │ │ ├── necks │ │ │ │ ├── __init__.py │ │ │ │ ├── featurepyramid.py │ │ │ │ ├── fpn.py │ │ │ │ ├── ic_neck.py │ │ │ │ ├── jpu.py │ │ │ │ ├── mla_neck.py │ │ │ │ └── multilevel_neck.py │ │ │ ├── segmentors │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── cascade_encoder_decoder.py │ │ │ │ └── encoder_decoder.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── embed.py │ │ │ │ ├── inverted_residual.py │ │ │ │ ├── make_divisible.py │ │ │ │ ├── res_layer.py │ │ │ │ ├── se_layer.py │ │ │ │ ├── self_attention_block.py │ │ │ │ ├── shape_convert.py │ │ │ │ └── up_conv_block.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── encoding.py │ │ │ └── wrappers.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── collect_env.py │ │ │ ├── logger.py │ │ │ ├── misc.py │ │ │ ├── set_env.py │ │ │ └── util_distribution.py │ │ └── version.py │ ├── model-index.yml │ ├── pytest.ini │ ├── requirements.txt │ ├── requirements │ │ ├── docs.txt │ │ ├── mminstall.txt │ │ ├── optional.txt │ │ ├── readthedocs.txt │ │ ├── runtime.txt │ │ └── tests.txt │ ├── resources │ │ ├── 3dogs.jpg │ │ ├── 3dogs_mask.png │ │ ├── mmseg-logo.png │ │ └── seg_demo.gif │ ├── setup.cfg │ ├── setup.py │ ├── tests │ │ ├── __init__.py │ │ ├── test_apis │ │ │ └── test_single_gpu.py │ │ ├── test_config.py │ │ ├── test_core │ │ │ ├── test_layer_decay_optimizer_constructor.py │ │ │ └── test_optimizer.py │ │ ├── test_data │ │ │ ├── test_dataset.py │ │ │ ├── test_dataset_builder.py │ │ │ ├── test_loading.py │ │ │ ├── test_transform.py │ │ │ └── test_tta.py │ │ ├── test_digit_version.py │ │ ├── test_eval_hook.py │ │ ├── test_inference.py │ │ ├── test_metrics.py │ │ ├── test_models │ │ │ ├── __init__.py │ │ │ ├── test_backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── test_beit.py │ │ │ │ ├── test_bisenetv1.py │ │ │ │ ├── test_bisenetv2.py │ │ │ │ ├── test_blocks.py │ │ │ │ ├── test_cgnet.py │ │ │ │ ├── test_erfnet.py │ │ │ │ ├── test_fast_scnn.py │ │ │ │ ├── test_hrnet.py │ │ │ │ ├── test_icnet.py │ │ │ │ ├── test_mae.py │ │ │ │ ├── test_mit.py │ │ │ │ ├── test_mobilenet_v3.py │ │ │ │ ├── test_resnest.py │ │ │ │ ├── test_resnet.py │ │ │ │ ├── test_resnext.py │ │ │ │ ├── test_stdc.py │ │ │ │ ├── test_swin.py │ │ │ │ ├── test_timm_backbone.py │ │ │ │ ├── test_twins.py │ │ │ │ ├── test_unet.py │ │ │ │ ├── test_vit.py │ │ │ │ └── utils.py │ │ │ ├── test_forward.py │ │ │ ├── test_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── test_ann_head.py │ │ │ │ ├── test_apc_head.py │ │ │ │ ├── test_aspp_head.py │ │ │ │ ├── test_cc_head.py │ │ │ │ ├── test_da_head.py │ │ │ │ ├── test_decode_head.py │ │ │ │ ├── test_dm_head.py │ │ │ │ ├── test_dnl_head.py │ │ │ │ ├── test_dpt_head.py │ │ │ │ ├── test_ema_head.py │ │ │ │ ├── test_enc_head.py │ │ │ │ ├── test_fcn_head.py │ │ │ │ ├── test_gc_head.py │ │ │ │ ├── test_isa_head.py │ │ │ │ ├── test_knet_head.py │ │ │ │ ├── test_lraspp_head.py │ │ │ │ ├── test_nl_head.py │ │ │ │ ├── test_ocr_head.py │ │ │ │ ├── test_point_head.py │ │ │ │ ├── test_psa_head.py │ │ │ │ ├── test_psp_head.py │ │ │ │ ├── test_segformer_head.py │ │ │ │ ├── test_segmenter_mask_head.py │ │ │ │ ├── test_setr_mla_head.py │ │ │ │ ├── test_setr_up_head.py │ │ │ │ ├── test_stdc_head.py │ │ │ │ ├── test_uper_head.py │ │ │ │ └── utils.py │ │ │ ├── test_losses │ │ │ │ ├── __init__.py │ │ │ │ ├── test_ce_loss.py │ │ │ │ ├── test_dice_loss.py │ │ │ │ ├── test_focal_loss.py │ │ │ │ ├── test_lovasz_loss.py │ │ │ │ ├── test_tversky_loss.py │ │ │ │ └── test_utils.py │ │ │ ├── test_necks │ │ │ │ ├── __init__.py │ │ │ │ ├── test_feature2pyramid.py │ │ │ │ ├── test_fpn.py │ │ │ │ ├── test_ic_neck.py │ │ │ │ ├── test_jpu.py │ │ │ │ ├── test_mla_neck.py │ │ │ │ └── test_multilevel_neck.py │ │ │ ├── test_segmentors │ │ │ │ ├── __init__.py │ │ │ │ ├── test_cascade_encoder_decoder.py │ │ │ │ ├── test_encoder_decoder.py │ │ │ │ └── utils.py │ │ │ └── test_utils │ │ │ │ ├── __init__.py │ │ │ │ ├── test_embed.py │ │ │ │ └── test_shape_convert.py │ │ ├── test_sampler.py │ │ └── test_utils │ │ │ ├── test_misc.py │ │ │ ├── test_set_env.py │ │ │ └── test_util_distribution.py │ └── tools │ │ ├── analyze_logs.py │ │ ├── benchmark.py │ │ ├── browse_dataset.py │ │ ├── confusion_matrix.py │ │ ├── convert_datasets │ │ ├── chase_db1.py │ │ ├── cityscapes.py │ │ ├── coco_stuff10k.py │ │ ├── coco_stuff164k.py │ │ ├── drive.py │ │ ├── hrf.py │ │ ├── isaid.py │ │ ├── loveda.py │ │ ├── pascal_context.py │ │ ├── potsdam.py │ │ ├── stare.py │ │ ├── vaihingen.py │ │ └── voc_aug.py │ │ ├── deploy_test.py │ │ ├── dist_test.sh │ │ ├── dist_train.sh │ │ ├── get_flops.py │ │ ├── model_converters │ │ ├── beit2mmseg.py │ │ ├── mit2mmseg.py │ │ ├── stdc2mmseg.py │ │ ├── swin2mmseg.py │ │ ├── twins2mmseg.py │ │ ├── vit2mmseg.py │ │ └── vitjax2mmseg.py │ │ ├── model_ensemble.py │ │ ├── onnx2tensorrt.py │ │ ├── print_config.py │ │ ├── publish_model.py │ │ ├── pytorch2onnx.py │ │ ├── pytorch2torchscript.py │ │ ├── slurm_test.sh │ │ ├── slurm_train.sh │ │ ├── test.py │ │ ├── torchserve │ │ ├── mmseg2torchserve.py │ │ ├── mmseg_handler.py │ │ └── test_torchserve.py │ │ └── train.py ├── scripts │ ├── eval_vim_small_upernet.sh │ ├── eval_vim_tiny_upernet.sh │ ├── ft_vim_small_upernet.sh │ └── ft_vim_tiny_upernet.sh ├── seg-requirements.txt ├── test.py ├── tools │ └── analysis_tools │ │ ├── benchmark.py │ │ ├── benchmark_scripts.sh │ │ ├── deit-t-upernet.txt │ │ ├── plot_curve.ipynb │ │ └── vim-t-upernet.txt └── train.py └── vim ├── .gitignore ├── LICENSE ├── augment.py ├── datasets.py ├── dyvm.py ├── engine.py ├── hubconf.py ├── losses.py ├── main.py ├── models_mamba.py ├── rope.py ├── run_with_submitit.py ├── samplers.py ├── scripts ├── ft-vim-s.sh ├── ft-vim-t.sh ├── pt-vim-s.sh └── pt-vim-t.sh ├── utils.py └── vim_requirements.txt /Ablation/Block_Only/mamba-1p1p1/AUTHORS: -------------------------------------------------------------------------------- 1 | Tri Dao, tri@tridao.me 2 | Albert Gu, agu@andrew.cmu.edu 3 | -------------------------------------------------------------------------------- /Ablation/Block_Only/mamba-1p1p1/mamba_ssm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Block_Only/mamba-1p1p1/mamba_ssm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Block_Only/mamba-1p1p1/mamba_ssm/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Block_Only/mamba-1p1p1/mamba_ssm/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Block_Only/mamba-1p1p1/mamba_ssm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/=1.1.0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/docs/.gitignore: -------------------------------------------------------------------------------- 1 | _build 2 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/docs/notes/contributing.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/docs/tutorials/builtin_datasets.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/docs/tutorials/getting_started.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/docs/tutorials/install.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/projects/DensePose/densepose/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/projects/DensePose/densepose/vis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/tests/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/tests/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/tests/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/tests/structures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/tests/tracking/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/det/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/mamba-1p1p1/mamba_ssm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/mamba-1p1p1/mamba_ssm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/mamba-1p1p1/mamba_ssm/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/mamba-1p1p1/mamba_ssm/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/mamba-1p1p1/mamba_ssm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-B/vim/=1.8.0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/=1.1.0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/docs/.gitignore: -------------------------------------------------------------------------------- 1 | _build 2 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/docs/notes/contributing.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/docs/tutorials/builtin_datasets.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/docs/tutorials/getting_started.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/docs/tutorials/install.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/projects/DensePose/densepose/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/projects/DensePose/densepose/vis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/tests/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/tests/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/tests/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/tests/structures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/tests/tracking/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/det/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/mamba-1p1p1/mamba_ssm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/mamba-1p1p1/mamba_ssm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/mamba-1p1p1/mamba_ssm/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/mamba-1p1p1/mamba_ssm/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/mamba-1p1p1/mamba_ssm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Predictor_Input/Predictor-delta/vim/=1.8.0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/=1.1.0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/docs/.gitignore: -------------------------------------------------------------------------------- 1 | _build 2 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/docs/notes/contributing.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/docs/tutorials/builtin_datasets.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/docs/tutorials/getting_started.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/docs/tutorials/install.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/projects/DensePose/densepose/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/projects/DensePose/densepose/vis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/tests/config/dir1/bad_import2.py: -------------------------------------------------------------------------------- 1 | from .does_not_exist import x 2 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/tests/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/tests/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/tests/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/tests/structures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/tests/tracking/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/det/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/mamba-1p1p1/mamba_ssm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/mamba-1p1p1/mamba_ssm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/mamba-1p1p1/mamba_ssm/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/mamba-1p1p1/mamba_ssm/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/mamba-1p1p1/mamba_ssm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Random/vim/=1.8.0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/=1.1.0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/docs/.gitignore: -------------------------------------------------------------------------------- 1 | _build 2 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/docs/notes/contributing.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/docs/tutorials/builtin_datasets.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/docs/tutorials/getting_started.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/docs/tutorials/install.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/projects/DensePose/densepose/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/projects/DensePose/densepose/vis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/tests/config/dir1/bad_import2.py: -------------------------------------------------------------------------------- 1 | from .does_not_exist import x 2 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/tests/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/tests/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/tests/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/tests/structures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/tests/tracking/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/det/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/mamba-1p1p1/mamba_ssm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/mamba-1p1p1/mamba_ssm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/mamba-1p1p1/mamba_ssm/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/mamba-1p1p1/mamba_ssm/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/mamba-1p1p1/mamba_ssm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Pruning_Strategy/Static/vim/=1.8.0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Token_Only/mamba-1p1p1/AUTHORS: -------------------------------------------------------------------------------- 1 | Tri Dao, tri@tridao.me 2 | Albert Gu, agu@andrew.cmu.edu 3 | -------------------------------------------------------------------------------- /Ablation/Token_Only/mamba-1p1p1/mamba_ssm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Token_Only/mamba-1p1p1/mamba_ssm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Token_Only/mamba-1p1p1/mamba_ssm/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Token_Only/mamba-1p1p1/mamba_ssm/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/Token_Only/mamba-1p1p1/mamba_ssm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Ablation/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/DyVM/HEAD/Ablation/readme.md -------------------------------------------------------------------------------- /Asset/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/DyVM/HEAD/Asset/pipeline.png -------------------------------------------------------------------------------- /Image_Classification/MambaReg/causal-conv1d/AUTHORS: -------------------------------------------------------------------------------- 1 | Tri Dao, tri@tridao.me 2 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/causal-conv1d/causal_conv1d.egg-info/dependency_links.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/mamba-1p1p1/mamba_ssm.egg-info/dependency_links.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/mamba-1p1p1/mamba_ssm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/mamba-1p1p1/mamba_ssm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/mamba-1p1p1/mamba_ssm/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/mamba-1p1p1/mamba_ssm/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/mamba-1p1p1/mamba_ssm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/ThreeAugment: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/batch_size: -------------------------------------------------------------------------------- 1 | 64 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/block_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/cooldown_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/cutmix_minmax: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/dist_backend: -------------------------------------------------------------------------------- 1 | nccl -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/distillation_tau: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/eval: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/eval_crop_ratio: -------------------------------------------------------------------------------- 1 | 0.875 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/if_continue_inf: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/inat_category: -------------------------------------------------------------------------------- 1 | name -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/layer_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.7 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/lr: -------------------------------------------------------------------------------- 1 | 5e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/lr_noise_pct: -------------------------------------------------------------------------------- 1 | 0.67 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/mixup_switch_prob: -------------------------------------------------------------------------------- 1 | 0.5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/model: -------------------------------------------------------------------------------- 1 | Vim-Small -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/patience_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/repeated_aug: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/token_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.7 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/weight_decay: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/tags/mlflow.runName: -------------------------------------------------------------------------------- 1 | vim-s -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/029766d5a88749ad83a0cb417b2b0bae/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/ThreeAugment: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/batch_size: -------------------------------------------------------------------------------- 1 | 64 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/block_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/cooldown_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/cutmix_minmax: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/dist_backend: -------------------------------------------------------------------------------- 1 | nccl -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/distillation_tau: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/eval_crop_ratio: -------------------------------------------------------------------------------- 1 | 0.875 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/if_continue_inf: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/inat_category: -------------------------------------------------------------------------------- 1 | name -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/layer_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.7 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/lr: -------------------------------------------------------------------------------- 1 | 3e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/lr_noise_pct: -------------------------------------------------------------------------------- 1 | 0.67 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/mixup_switch_prob: -------------------------------------------------------------------------------- 1 | 0.5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/model: -------------------------------------------------------------------------------- 1 | LocalMamba-Tiny -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/n_parameters: -------------------------------------------------------------------------------- 1 | 8237838 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/patience_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/repeated_aug: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/token_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/weight_decay: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fb7b6a0f7e4880bab6c032bd77e973/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/ThreeAugment: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/batch_size: -------------------------------------------------------------------------------- 1 | 64 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/block_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/cooldown_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/cutmix_minmax: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/dist_backend: -------------------------------------------------------------------------------- 1 | nccl -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/distillation_tau: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/eval_crop_ratio: -------------------------------------------------------------------------------- 1 | 0.875 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/if_continue_inf: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/inat_category: -------------------------------------------------------------------------------- 1 | name -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/layer_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.7 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/lr: -------------------------------------------------------------------------------- 1 | 3e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/lr_noise_pct: -------------------------------------------------------------------------------- 1 | 0.67 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/mixup_switch_prob: -------------------------------------------------------------------------------- 1 | 0.5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/model: -------------------------------------------------------------------------------- 1 | LocalMamba-Tiny -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/patience_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/repeated_aug: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/token_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/weight_decay: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/params/world_size: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/24fcf6845c004274bd681e7d3851c721/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/ThreeAugment: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/batch_size: -------------------------------------------------------------------------------- 1 | 32 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/block_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/cooldown_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/cutmix_minmax: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/dist_backend: -------------------------------------------------------------------------------- 1 | nccl -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/distillation_tau: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/eval_crop_ratio: -------------------------------------------------------------------------------- 1 | 0.875 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/if_continue_inf: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/inat_category: -------------------------------------------------------------------------------- 1 | name -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/layer_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.7 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/lr: -------------------------------------------------------------------------------- 1 | 3e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/lr_noise_pct: -------------------------------------------------------------------------------- 1 | 0.67 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/mixup_switch_prob: -------------------------------------------------------------------------------- 1 | 0.5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/model: -------------------------------------------------------------------------------- 1 | LocalMamba-Tiny -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/n_parameters: -------------------------------------------------------------------------------- 1 | 8237838 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/patience_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/repeated_aug: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/token_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/weight_decay: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/2bb5ad6028374f0bb8f1b97c9a55a60e/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/ThreeAugment: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/batch_size: -------------------------------------------------------------------------------- 1 | 48 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/block_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/cooldown_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/cutmix_minmax: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/dist_backend: -------------------------------------------------------------------------------- 1 | nccl -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/distillation_tau: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/eval_crop_ratio: -------------------------------------------------------------------------------- 1 | 0.875 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/if_continue_inf: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/inat_category: -------------------------------------------------------------------------------- 1 | name -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/layer_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.7 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/lr: -------------------------------------------------------------------------------- 1 | 3e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/lr_noise_pct: -------------------------------------------------------------------------------- 1 | 0.67 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/mixup_switch_prob: -------------------------------------------------------------------------------- 1 | 0.5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/model: -------------------------------------------------------------------------------- 1 | LocalMamba-Tiny -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/n_parameters: -------------------------------------------------------------------------------- 1 | 8237838 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/patience_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/repeated_aug: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/token_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/weight_decay: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/3ab1b30b4ed64e709f0333313df49100/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/ThreeAugment: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/batch_size: -------------------------------------------------------------------------------- 1 | 64 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/block_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/cooldown_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/cutmix_minmax: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/dist_backend: -------------------------------------------------------------------------------- 1 | nccl -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/distillation_tau: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/eval: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/eval_crop_ratio: -------------------------------------------------------------------------------- 1 | 0.875 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/if_continue_inf: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/inat_category: -------------------------------------------------------------------------------- 1 | name -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/layer_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.7 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/lr: -------------------------------------------------------------------------------- 1 | 5e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/lr_noise_pct: -------------------------------------------------------------------------------- 1 | 0.67 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/mixup_switch_prob: -------------------------------------------------------------------------------- 1 | 0.5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/model: -------------------------------------------------------------------------------- 1 | MambaReg-Small -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/patience_epochs: -------------------------------------------------------------------------------- 1 | 10 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/repeated_aug: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/token_keep_ratio: -------------------------------------------------------------------------------- 1 | 0.7 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/params/world_size: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/5f3fa3b2ba5c4603ba890982f3e7daa0/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/batch_size: -------------------------------------------------------------------------------- 1 | 32 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/eval: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/lr: -------------------------------------------------------------------------------- 1 | 5e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/model: -------------------------------------------------------------------------------- 1 | Vim-Base -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/66460dbc9d1f4ae2b174a274e1213147/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/batch_size: -------------------------------------------------------------------------------- 1 | 64 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/lr: -------------------------------------------------------------------------------- 1 | 3e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/87d8e3915a02419eaf8c830247982552/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/batch_size: -------------------------------------------------------------------------------- 1 | 48 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/lr: -------------------------------------------------------------------------------- 1 | 3e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9bc80b4c33f647dc98b31985835ec501/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/batch_size: -------------------------------------------------------------------------------- 1 | 64 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/eval: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/lr: -------------------------------------------------------------------------------- 1 | 5e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/params/world_size: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/9e3fd56185184925bb8823ce08931f33/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/batch_size: -------------------------------------------------------------------------------- 1 | 24 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/lr: -------------------------------------------------------------------------------- 1 | 3e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/b41b98c70a4544a699ff1b76a77e7ffe/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/batch_size: -------------------------------------------------------------------------------- 1 | 64 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/eval: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/lr: -------------------------------------------------------------------------------- 1 | 5e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/model: -------------------------------------------------------------------------------- 1 | Vim-Small -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/bb60d4d2bd004e2cb692dee5a73028b5/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/attn_only: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/batch_size: -------------------------------------------------------------------------------- 1 | 64 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/bce_loss: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/clip_grad: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/color_jitter: -------------------------------------------------------------------------------- 1 | 0.3 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/cosub: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/cutmix: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/data_set: -------------------------------------------------------------------------------- 1 | IMNET -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/decay_epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/decay_rate: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/device: -------------------------------------------------------------------------------- 1 | cuda -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/dist_eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/dist_url: -------------------------------------------------------------------------------- 1 | env:// -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/distributed: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/drop: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/drop_path: -------------------------------------------------------------------------------- 1 | 0.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/epochs: -------------------------------------------------------------------------------- 1 | 30 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/eval: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/finetune: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/gpu: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/if_amp: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/if_nan2num: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/input_size: -------------------------------------------------------------------------------- 1 | 224 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/local_rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/lr: -------------------------------------------------------------------------------- 1 | 5e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/lr_noise: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/lr_noise_std: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/min_lr: -------------------------------------------------------------------------------- 1 | 1e-06 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/mixup: -------------------------------------------------------------------------------- 1 | 0.8 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/mixup_mode: -------------------------------------------------------------------------------- 1 | batch -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/mixup_prob: -------------------------------------------------------------------------------- 1 | 1.0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/model_ema: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/momentum: -------------------------------------------------------------------------------- 1 | 0.9 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/num_workers: -------------------------------------------------------------------------------- 1 | 16 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/opt: -------------------------------------------------------------------------------- 1 | adamw -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/opt_betas: -------------------------------------------------------------------------------- 1 | None -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/opt_eps: -------------------------------------------------------------------------------- 1 | 1e-08 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/pin_mem: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/rank: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/recount: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/remode: -------------------------------------------------------------------------------- 1 | pixel -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/reprob: -------------------------------------------------------------------------------- 1 | 0.25 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/resplit: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/resume: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/sched: -------------------------------------------------------------------------------- 1 | cosine -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/seed: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/smoothing: -------------------------------------------------------------------------------- 1 | 0.1 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/src: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/start_epoch: -------------------------------------------------------------------------------- 1 | 0 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/train_mode: -------------------------------------------------------------------------------- 1 | True -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/unscale_lr: -------------------------------------------------------------------------------- 1 | False -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/warmup_epochs: -------------------------------------------------------------------------------- 1 | 5 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/warmup_lr: -------------------------------------------------------------------------------- 1 | 1e-05 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/params/world_size: -------------------------------------------------------------------------------- 1 | 4 -------------------------------------------------------------------------------- /Image_Classification/MambaReg/vim/mlruns/0/c46388690df848a481bc263fb15306e4/tags/mlflow.user: -------------------------------------------------------------------------------- 1 | root -------------------------------------------------------------------------------- /Image_Classification/Vim&VideoMamba/causal-conv1d/AUTHORS: -------------------------------------------------------------------------------- 1 | Tri Dao, tri@tridao.me 2 | -------------------------------------------------------------------------------- /Image_Classification/Vim&VideoMamba/mamba-1p1p1/mamba_ssm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/Vim&VideoMamba/mamba-1p1p1/mamba_ssm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/Vim&VideoMamba/mamba-1p1p1/mamba_ssm/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/Vim&VideoMamba/mamba-1p1p1/mamba_ssm/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Image_Classification/Vim&VideoMamba/mamba-1p1p1/mamba_ssm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/DyVM/HEAD/README.md -------------------------------------------------------------------------------- /Semantic_Segmentation/mamba-1p1p1/mamba_ssm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Semantic_Segmentation/mamba-1p1p1/mamba_ssm/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Semantic_Segmentation/mamba-1p1p1/mamba_ssm/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Semantic_Segmentation/mamba-1p1p1/mamba_ssm/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Semantic_Segmentation/mamba-1p1p1/mamba_ssm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Semantic_Segmentation/seg/mmsegmentation/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | cityscapesscripts 2 | --------------------------------------------------------------------------------