├── .gitignore ├── .pre-commit-config.yaml ├── .style.yapf ├── LICENSE ├── README.md ├── README_cn.md ├── codestyle ├── .gitignore ├── clang_format.hook ├── copyright.hook ├── cpplint_pre_commit.hook ├── docstring_checker.py ├── pylint_pre_commit.hook └── test_docstring_checker.py ├── configs ├── beit │ ├── README.md │ ├── beit_base_p16_224.yaml │ ├── beit_base_p16_224_ft_1k.yaml │ ├── beit_base_p16_224_pt_1k.yaml │ ├── beit_base_p16_384.yaml │ ├── beit_large_p16_224.yaml │ ├── beit_large_p16_384.yaml │ └── beit_large_p16_512.yaml ├── byol │ ├── byol_clas_r50.yaml │ └── byol_r50_IM.yaml ├── cait │ ├── README.md │ ├── cait_m36_384.yaml │ ├── cait_m48_448.yaml │ ├── cait_s24_224.yaml │ ├── cait_s24_384.yaml │ ├── cait_s36_384.yaml │ └── cait_xs24_384.yaml ├── clip │ └── vit-b-32.yaml ├── convnext │ ├── README.md │ ├── convnext_small_224.yaml │ └── convnext_tiny_224.yaml ├── cvt │ ├── README.md │ ├── cvt_13_224.yaml │ ├── cvt_13_384.yaml │ ├── cvt_21_224.yaml │ ├── cvt_21_384.yaml │ └── cvt_w24_384.yaml ├── deit │ └── deit-base-p16-pt_in1k-224_2n16c_fp16_o1_dp.yaml ├── densecl │ ├── README.md │ ├── README_ch.md │ └── densecl_r50.yaml ├── lvvit │ ├── README.md │ ├── README_ch.md │ └── lvvit_tiny.yaml ├── mae │ ├── mae_vit_b_finetune.yaml │ └── mae_vit_b_pretrain.yaml ├── mlp_mixer │ ├── README.md │ ├── mlp-mixer_b16_224.yaml │ └── mlp-mixer_l16_224.yaml ├── moco │ ├── moco_clas_r50.yaml │ ├── moco_v1_r50.yaml │ └── moco_v2_r50.yaml ├── moco_byol │ ├── moco_byol_r50_IM.yaml │ └── moco_byol_r50_IM_clas.yaml ├── pixpro │ ├── pixpro_base_r50_100ep.yaml │ └── pixpro_base_r50_100ep_IM_clas.yaml ├── simclr │ ├── simclr_r18_cifar10.yaml │ └── simclr_r50_IM.yaml ├── simsiam │ ├── README.md │ ├── README_ch.md │ ├── simsiam_clas_r50.yaml │ └── simsiam_r50.yaml ├── swav │ ├── README.md │ ├── README_ch.md │ ├── swav_clas_r50.yaml │ └── swav_r50_100ep.yaml ├── swin_transformer │ ├── SwinTransformer_base_patch4_window7_224.yaml │ ├── SwinTransformer_giant_patch4_window7_224.yaml │ ├── SwinTransformer_huge_patch4_window7_224.yaml │ ├── SwinTransformer_small_patch4_window7_224.yaml │ └── SwinTransformer_tiny_patch4_window7_224.yaml ├── t2t_vit │ ├── README.md │ ├── t2t_vit_14.yaml │ ├── t2t_vit_19.yaml │ ├── t2t_vit_24.yaml │ ├── t2t_vit_t_14.yaml │ ├── t2t_vit_t_19.yaml │ └── t2t_vit_t_24.yaml ├── vision_transformer │ ├── vit-base-p16-ft_in1k-384.yaml │ ├── vit-base-p16-pt_in1k-224_4n32c_fp16_o1_dp.yaml │ ├── vit-base-p16-pt_in1k-224_4n32c_fp16_o1_sharding.yaml │ ├── vit-base-p16-pt_in1k-224_4n32c_fp16_o2_dp.yaml │ ├── vit-base-p16-pt_in1k-224_4n32c_fp16_o2_sharding.yaml │ ├── vit-base-p16-pt_in1k-224_4n32c_fp32_dp.yaml │ ├── vit-base-p32_ft_in1k-384.yaml │ └── vit-g-p14-pt_in1k-224_1n8c.yaml └── xcit │ ├── README.md │ ├── xcit_large_24_p16_224.yaml │ ├── xcit_large_24_p8_224.yaml │ ├── xcit_large_24_p8_224_dist.yaml │ ├── xcit_medium_24_p16_224.yaml │ ├── xcit_medium_24_p8_224.yaml │ ├── xcit_medium_24_p8_224_dist.yaml │ ├── xcit_nano_12_p16_224.yaml │ ├── xcit_nano_12_p8_224.yaml │ ├── xcit_nano_12_p8_224_dist.yaml │ ├── xcit_small_12_p16_224.yaml │ ├── xcit_small_12_p8_224.yaml │ ├── xcit_small_12_p8_224_dist.yaml │ ├── xcit_small_24_p16_224.yaml │ ├── xcit_small_24_p8_224.yaml │ ├── xcit_small_24_p8_224_dist.yaml │ ├── xcit_tiny_12_p16_224.yaml │ ├── xcit_tiny_12_p8_224.yaml │ ├── xcit_tiny_12_p8_224_dist.yaml │ ├── xcit_tiny_24_p16_224.yaml │ ├── xcit_tiny_24_p8_224.yaml │ └── xcit_tiny_24_p8_224_dist.yaml ├── deploy ├── configs │ └── inference_cls.yaml ├── images │ └── ILSVRC2012_val_00000010.jpeg ├── python │ ├── __init__.py │ ├── postprocess.py │ ├── predict_cls.py │ └── preprocess.py └── utils │ ├── __init__.py │ ├── config.py │ ├── get_image_list.py │ ├── logger.py │ └── predictor.py ├── docs ├── Classification_Models_Guide.md ├── GETTING_STARTED.md ├── INSTALL.md ├── Large_Scale_Training.md ├── Train_BYOL_model.md ├── Train_CLIP_model.md ├── Train_DenseCL_model.md ├── Train_MoCo-BYOL_model.md ├── Train_MoCo_model.md ├── Train_PixPro_model.md ├── Train_SimCLR_model.md ├── Train_SimSiam_model.md ├── Train_SwAV_model.md └── imgs │ ├── clip_demo1.png │ ├── clip_demo2.png │ ├── densecl.png │ ├── moco-byol.png │ ├── passl_logo.svg │ ├── simsiam.png │ └── swav.png ├── passl ├── __init__.py ├── core │ ├── __init__.py │ ├── grad_clip.py │ ├── grad_scaler.py │ ├── param_fuse.py │ ├── recompute.py │ └── sync_utils.py ├── data │ ├── __init__.py │ ├── dataset │ │ ├── __init__.py │ │ ├── common_dataset.py │ │ ├── fewshot_dataset.py │ │ ├── imagefolder_dataset.py │ │ ├── imagenet_dataset.py │ │ └── swavmulticrop_datatset.py │ ├── postprocess │ │ └── __init__.py │ ├── preprocess │ │ ├── __init__.py │ │ ├── basic_transforms.py │ │ ├── batch_transforms.py │ │ └── timm_autoaugment.py │ ├── sampler │ │ ├── __init__.py │ │ └── repeatedaug_sampler.py │ └── utils │ │ ├── __init__.py │ │ ├── batch_collate_fn.py │ │ └── transform_utils.py ├── distributed │ ├── __init__.py │ ├── env.py │ └── nn │ │ ├── __init__.py │ │ └── functional.py ├── engine │ ├── __init__.py │ ├── engine.py │ ├── inference.py │ └── loops │ │ ├── __init__.py │ │ ├── classification_loop.py │ │ ├── contrastive_learning_loop.py │ │ └── loop.py ├── loss │ ├── __init__.py │ └── celoss.py ├── metric │ ├── __init__.py │ └── metrics.py ├── models │ ├── __init__.py │ ├── base_model.py │ ├── cae.py │ ├── cait.py │ ├── convmae │ │ ├── __init__.py │ │ ├── conv_mae.py │ │ └── conv_vit.py │ ├── convnext.py │ ├── deit.py │ ├── dino │ │ ├── __init__.py │ │ └── dino_vit.py │ ├── dinov2 │ │ ├── __init__.py │ │ └── dinov2_vit.py │ ├── mae.py │ ├── mocov3.py │ ├── resnet.py │ ├── simsiam.py │ ├── swav.py │ ├── swin_transformer.py │ ├── utils │ │ ├── __init__.py │ │ ├── averaged_model.py │ │ ├── ema.py │ │ ├── pos_embed.py │ │ └── tome.py │ ├── vision_transformer.py │ └── vision_transformer_hybrid.py ├── nn │ ├── __init__.py │ ├── finer_grained_mp_layers.py │ ├── init.py │ └── norm.py ├── optimizer │ ├── __init__.py │ ├── adafactor.py │ ├── adamw.py │ ├── adan.py │ ├── momentum.py │ ├── momentum_larc.py │ ├── momentum_lars.py │ ├── optimizer.py │ └── utils │ │ ├── __init__.py │ │ └── group_params.py ├── scheduler │ ├── __init__.py │ ├── lr_callable.py │ └── lr_scheduler.py └── utils │ ├── __init__.py │ ├── config.py │ ├── infohub.py │ ├── io.py │ ├── logger.py │ ├── misc.py │ └── profiler.py ├── passl_v110 ├── __init__.py ├── datasets │ ├── __init__.py │ ├── beitdataset.py │ ├── builder.py │ ├── cifar.py │ ├── folder.py │ ├── imagenet.py │ ├── lvdataset.py │ ├── multicropdataset.py │ ├── preprocess │ │ ├── __init__.py │ │ ├── auto_augment.py │ │ ├── builder.py │ │ ├── constants.py │ │ ├── cv2_func.py │ │ ├── cv2_trans.py │ │ ├── img_pil_pixpro_transforms.py │ │ ├── lvvit │ │ │ ├── __init__.py │ │ │ ├── auto_augment.py │ │ │ ├── mixup.py │ │ │ └── transforms.py │ │ ├── masking_generator.py │ │ ├── mixup.py │ │ ├── random_erasing.py │ │ └── transforms.py │ └── textimagedataset.py ├── engine │ ├── __init__.py │ └── trainer.py ├── hooks │ ├── __init__.py │ ├── builder.py │ ├── checkpoint_hook.py │ ├── evaluate_hook.py │ ├── hook.py │ ├── log_hook.py │ ├── lr_scheduler_hook.py │ ├── optimizer_hook.py │ ├── timer_hook.py │ └── visual_hook.py ├── modeling │ ├── __init__.py │ ├── architectures │ │ ├── BEiTWrapper.py │ │ ├── BYOL.py │ │ ├── CLIPWrapper.py │ │ ├── CaiTWrapper.py │ │ ├── CvTWrapper.py │ │ ├── DeiTWrapper.py │ │ ├── DistillationWrapper.py │ │ ├── LVViTWrapper.py │ │ ├── MAE.py │ │ ├── MlpMixerWrapper.py │ │ ├── MoCoBYOL.py │ │ ├── SwinWrapper.py │ │ ├── T2TViTWrapper.py │ │ ├── ViTWrapper.py │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── byol_clas.py │ │ ├── clas.py │ │ ├── densecl.py │ │ ├── moco.py │ │ ├── pixpro.py │ │ ├── simclr.py │ │ ├── simsiam.py │ │ └── swav.py │ ├── backbones │ │ ├── __init__.py │ │ ├── base_transformer.py │ │ ├── beit.py │ │ ├── beit_ft.py │ │ ├── beit_pt.py │ │ ├── builder.py │ │ ├── cait.py │ │ ├── clip.py │ │ ├── convnext.py │ │ ├── cvt.py │ │ ├── deit.py │ │ ├── discrete_vae.py │ │ ├── lvvit.py │ │ ├── mae.py │ │ ├── mlp_mixer.py │ │ ├── regnet.py │ │ ├── resnet.py │ │ ├── resnetcifar.py │ │ ├── resnetimagenet.py │ │ ├── resnetsimclr.py │ │ ├── resnetswav.py │ │ ├── swin_transformer.py │ │ ├── t2t_vit.py │ │ ├── transformer.py │ │ ├── vision_transformer.py │ │ ├── vit.py │ │ └── xcit.py │ ├── heads │ │ ├── __init__.py │ │ ├── beit_head.py │ │ ├── builder.py │ │ ├── cait_head.py │ │ ├── clas_head.py │ │ ├── clip_head.py │ │ ├── contrastive_head.py │ │ ├── cvt_head.py │ │ ├── l2_head.py │ │ ├── mb_head.py │ │ ├── mlp_mixer_head.py │ │ ├── pixpro_head.py │ │ ├── simclr_contrastive_head.py │ │ ├── simsiam_contrastive_head.py │ │ ├── swav_head.py │ │ ├── swin_transformer_head.py │ │ ├── t2t_vit_head.py │ │ └── vision_transformer_head.py │ └── necks │ │ ├── __init__.py │ │ ├── base_neck.py │ │ └── builder.py ├── modules │ ├── __init__.py │ ├── freeze.py │ ├── get_sincos_pe.py │ └── init.py ├── solver │ ├── __init__.py │ ├── builder.py │ ├── lr_scheduler.py │ └── optimizer.py └── utils │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── config.py │ ├── imagenet1k_label_list.txt │ ├── logger.py │ ├── misc.py │ ├── options.py │ ├── registry.py │ ├── setup.py │ └── tokenizer.py ├── requirements.txt ├── setup.py ├── tasks ├── accelerate │ └── tome │ │ ├── README.md │ │ └── validation_tome_vit.ipynb ├── classification │ ├── cait │ │ ├── README.md │ │ ├── configs │ │ │ └── cait_s24_224_in1k_1n8c_dp_fp16o2.yaml │ │ └── train.sh │ ├── convnext │ │ ├── README.md │ │ └── configs │ │ │ ├── ConvNeXt_base_224_in1k_1n8c_dp_fp32.yaml │ │ │ ├── ConvNeXt_base_224_in1k_4n32c_dp_fp16o2.yaml │ │ │ └── ConvNeXt_base_224_in1k_4n32c_dp_fp32.yaml │ ├── deit │ │ ├── README.md │ │ └── configs │ │ │ ├── DeiT_base_patch16_224_in1k_1n8c_dp_fp16o2.yaml │ │ │ ├── DeiT_base_patch16_224_in1k_1n8c_dp_fp32.yaml │ │ │ └── DeiT_base_patch16_224_in1k_2n16c_dp_fp16o2.yaml │ ├── swin │ │ ├── README.md │ │ ├── configs │ │ │ ├── swin_base_patch4_window7_224_in1k_1n8c_dp_fp16o1.yaml │ │ │ └── swin_base_patch4_window7_224_in1k_1n8c_dp_fp16o2.yaml │ │ └── train.sh │ └── vit │ │ ├── README.md │ │ ├── configs │ │ ├── ViT_base_patch16_224_in1k_1n8c_dp_fp16o2.yaml │ │ ├── ViT_base_patch16_224_in1k_adan_1n8c_dp_fp16o2.yaml │ │ ├── ViT_base_patch16_384_ft_in1k_1n8c_dp_fp16o2.yaml │ │ ├── ViT_large_patch16_224_in21k_4n32c_dp_fp16o2.yaml │ │ └── ViT_large_patch16_384_in1k_ft_4n32c_dp_fp16o2.yaml │ │ ├── eval.sh │ │ └── evaluate.py └── ssl │ ├── cae │ ├── README.md │ ├── dall_e │ │ ├── __init__.py │ │ ├── dalle_vae.py │ │ ├── decoder.py │ │ ├── encoder.py │ │ └── utils.py │ ├── engine_finetune.py │ ├── engine_pretrain.py │ ├── extract_model.py │ ├── finetune.sh │ ├── linprobe.sh │ ├── main_finetune.py │ ├── main_linprobe.py │ ├── main_pretrain.py │ ├── pretrain.sh │ └── util │ │ ├── loss.py │ │ ├── lr_decay.py │ │ ├── lr_sched.py │ │ ├── masking_generator.py │ │ └── misc.py │ ├── dino │ ├── README.md │ ├── configs │ │ ├── dino_deit_small_patch16_224_lp_in1k_1n8c_dp_fp16o1.yaml │ │ ├── dino_deit_small_patch8_224_lp_in1k_1n8c_dp_fp16o1.yaml │ │ ├── dino_vit_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.yaml │ │ └── dino_vit_base_patch8_224_lp_in1k_1n8c_dp_fp16o1.yaml │ └── linearprobe.sh │ ├── dinov2 │ ├── README.md │ ├── configs │ │ ├── dinov2_vit_base_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml │ │ ├── dinov2_vit_gaint2_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml │ │ ├── dinov2_vit_large_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml │ │ └── dinov2_vit_small_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml │ └── linearprobe.sh │ ├── mae │ ├── README.md │ ├── engine_finetune.py │ ├── engine_pretrain.py │ ├── finetune.sh │ ├── finetune_convmae.sh │ ├── linprobe.sh │ ├── linprobe_convmae.sh │ ├── main_finetune.py │ ├── main_linprobe.py │ ├── main_pretrain.py │ ├── pretrain.sh │ ├── pretrain_convmae.sh │ └── util │ │ ├── loss.py │ │ ├── lr_decay.py │ │ ├── lr_sched.py │ │ ├── misc.py │ │ ├── optim_factory.py │ │ └── pos_embed.py │ ├── mocov3 │ ├── README.md │ ├── builder_moco.py │ ├── configs │ │ ├── mocov3_deit_base_patch16_224_ft_in1k_1n8c_dp_fp16o1.yaml │ │ ├── mocov3_vit_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.yaml │ │ └── mocov3_vit_base_patch16_224_pt_in1k_4n32c_dp_fp16o1.yaml │ ├── extract_weight.py │ ├── finetune.sh │ ├── linearprobe.sh │ ├── main_lincls.py │ ├── main_moco.py │ ├── pretrain.sh │ └── vit_moco.py │ ├── simsiam │ ├── README.md │ ├── configs │ │ ├── simsiam_resnet50_lp_in1k_1n8c_dp_fp32.yaml │ │ └── simsiam_resnet50_pt_in1k_1n8c_dp_fp32.yaml │ ├── linearprobe.sh │ └── pretrain.sh │ └── swav │ ├── README.md │ ├── configs │ ├── swav_resnet50_224_ft_in1k_1n4c_dp_fp32.yaml │ ├── swav_resnet50_224_lp_in1k_1n8c_dp_fp32.yaml │ ├── swav_resnet50_224_pt_in1k_1n8c_dp_fp16o1.yaml │ └── swav_resnet50_224_pt_in1k_4n32c_dp_fp16o1.yaml │ ├── finetune.sh │ ├── linearprobe.sh │ ├── pretrain.sh │ └── pretrain_1N8C.sh ├── tests ├── CI │ ├── before_hook.sh │ ├── case.sh │ ├── classification │ │ ├── cait │ │ │ └── cait_s24_224_in1k_1n8c_dp_fp16o2.sh │ │ ├── convnext │ │ │ └── ConvNeXt_base_224_in1k_1n8c_dp_fp32.sh │ │ ├── deit │ │ │ ├── DeiT_base_patch16_224_in1k_1n8c_dp_fp16o2.sh │ │ │ └── DeiT_base_patch16_224_in1k_1n8c_dp_fp32.sh │ │ ├── swin │ │ │ └── swin_base_patch4_window7_224_fp16o2.sh │ │ └── vit │ │ │ ├── ViT_base_patch16_224_in1k_1n8c_dp_fp16o2.sh │ │ │ └── ViT_base_patch16_384_ft_in1k_1n8c_dp_fp16o2.sh │ ├── end_hook.sh │ ├── run_all.sh │ └── ssl │ │ ├── cae │ │ ├── cae_base_patch16_224_ft_in1k_1n8c_dp_fp16o1.sh │ │ ├── cae_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.sh │ │ └── cae_base_patch16_224_pt_in1k_1n8c_dp_fp16o1.sh │ │ ├── convmae │ │ ├── convmae_convvit_base_patch16_ft_in1k_1n8c_dp_fp16o1.sh │ │ ├── convmae_convvit_base_patch16_lp_in1k_1n8c_dp_fp16o1.sh │ │ └── convmae_convvit_base_patch16_pt_in1k_1n8c_dp_fp16o1.sh │ │ ├── dino │ │ └── dino_deit_small_patch16_224_lp_in1k_1n8c_dp_fp16o1.sh │ │ ├── dinov2 │ │ └── dinov2_vit_small_patch14_224_lp_in1k_1n8c_dp_fp16o1.sh │ │ ├── mae │ │ ├── mae_vit_base_patch16_ft_in1k_1n8c_dp_fp16o1.sh │ │ ├── mae_vit_base_patch16_lp_in1k_1n8c_dp_fp16o1.sh │ │ └── mae_vit_base_patch16_pt_in1k_1n8c_dp_fp16o1.sh │ │ ├── mocov3 │ │ ├── mocov3_deit_base_patch16_224_ft_in1k_1n8c_dp_fp16o1.sh │ │ ├── mocov3_vit_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.sh │ │ └── mocov3_vit_base_patch16_224_pt_in1k_1n8c_dp_fp16o1.sh │ │ ├── simsiam │ │ ├── simsiam_resnet50_lp_in1k_1n8c_dp_fp32.sh │ │ └── simsiam_resnet50_pt_in1k_1n8c_dp_fp32.sh │ │ └── swav │ │ ├── swav_resnet50_224_ft_in1k_1n4c_dp_fp32.sh │ │ ├── swav_resnet50_224_lp_in1k_1n8c_dp_fp32.sh │ │ └── swav_resnet50_224_pt_in1k_1n8c_dp_fp16o1.sh ├── test_tipc │ ├── README.md │ ├── classification │ │ ├── N1C8 │ │ │ ├── ConvNeXt_base_224_bs512_fp32_DP8-MP1.sh │ │ │ ├── DeiT_base_patch16_224_bs128_fp16o2_DP8-MP1.sh │ │ │ ├── DeiT_base_patch16_224_bs128_fp32_DP8-MP1.sh │ │ │ ├── ViT_base_patch16_224_bs512_fp16_DP8-MP1.sh │ │ │ ├── ViT_base_patch16_384_ft_bs512_fp16_DP8-MP1.sh │ │ │ ├── cait_s24_224_bs128_fp16o2_DP8-MP1.sh │ │ │ └── swin_base_patch4_window7_224_bs128_fp16o2_DP8-MP1.sh │ │ └── benchmark_common │ │ │ ├── prepare.sh │ │ │ └── run_benchmark.sh │ └── ssl │ │ ├── cae │ │ ├── N1C8 │ │ │ ├── cae_base_patch16_224_ft_bs128_fp16o1_DP8-MP1.sh │ │ │ ├── cae_base_patch16_224_lp_bs512_fp16o1_DP8-MP1.sh │ │ │ └── cae_base_patch16_224_pt_bs64_fp16o1_DP8-MP1.sh │ │ └── benchmark_common │ │ │ ├── prepare.sh │ │ │ └── run_benchmark.sh │ │ ├── convmae │ │ ├── N1C8 │ │ │ ├── convmae_convvit_base_patch16_ft_bs32_fp16o1_DP8-MP1.sh │ │ │ ├── convmae_convvit_base_patch16_lp_bs128_fp16o1_DP8-MP1.sh │ │ │ └── convmae_convvit_base_patch16_pt_bs64_fp16o1_DP8-MP1.sh │ │ └── benchmark_common │ │ │ ├── prepare.sh │ │ │ └── run_benchmark.sh │ │ ├── mae │ │ ├── N1C8 │ │ │ ├── mae_vit_base_patch16_ft_bs32_fp16o1_DP8-MP1.sh │ │ │ ├── mae_vit_base_patch16_lp_bs512_fp16o1_DP8-MP1.sh │ │ │ └── mae_vit_base_patch16_pt_bs128_fp16o1_DP8-MP1.sh │ │ └── benchmark_common │ │ │ ├── prepare.sh │ │ │ └── run_benchmark.sh │ │ ├── mocov3 │ │ ├── N1C8 │ │ │ ├── mocov3_deit_base_patch16_224_ft_bs128_fp16o1_DP8-MP1.sh │ │ │ ├── mocov3_vit_base_patch16_224_lp_bs128_fp16o1_DP8-MP1.sh │ │ │ └── mocov3_vit_base_patch16_224_pt_bs64_fp16o1_DP8-MP1.sh │ │ └── benchmark_common │ │ │ ├── prepare.sh │ │ │ └── run_benchmark.sh │ │ └── simsiam │ │ ├── N1C8 │ │ ├── simsiam_resnet50_lp_bs512_fp32_DP8-MP1.sh │ │ └── simsiam_resnet50_pt_bs64_fp32_DP8-MP1.sh │ │ └── benchmark_common │ │ ├── prepare.sh │ │ └── run_benchmark.sh └── test_tipc_v110 │ ├── README.md │ ├── common_func.sh │ ├── config │ ├── densecl_r50 │ │ └── train_infer_python.txt │ ├── lvvit │ │ └── train_infer_python.txt │ ├── moco_v1_r50 │ │ └── train_infer_python.txt │ ├── simsiam_r50 │ │ └── train_infer_python.txt │ └── swav_r50_100ep │ │ └── train_infer_python.txt │ ├── docs │ ├── guide.png │ ├── install.md │ └── test_train_inference_python.md │ ├── prepare.sh │ └── test_train_inference_python.sh ├── tools ├── eval.py ├── export.py └── train.py ├── tools_v110 ├── __init__.py ├── extract_weight.py ├── model_stat │ ├── __init__.py │ ├── get_flops.py │ ├── get_throughput.py │ └── op_count.py ├── passl2ppclas │ ├── convert.py │ └── ppclas_res50_keys.txt └── train.py └── version.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/.style.yapf -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/README.md -------------------------------------------------------------------------------- /README_cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/README_cn.md -------------------------------------------------------------------------------- /codestyle/.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | -------------------------------------------------------------------------------- /codestyle/clang_format.hook: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/codestyle/clang_format.hook -------------------------------------------------------------------------------- /codestyle/copyright.hook: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/codestyle/copyright.hook -------------------------------------------------------------------------------- /codestyle/cpplint_pre_commit.hook: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/codestyle/cpplint_pre_commit.hook -------------------------------------------------------------------------------- /codestyle/docstring_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/codestyle/docstring_checker.py -------------------------------------------------------------------------------- /codestyle/pylint_pre_commit.hook: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/codestyle/pylint_pre_commit.hook -------------------------------------------------------------------------------- /codestyle/test_docstring_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/codestyle/test_docstring_checker.py -------------------------------------------------------------------------------- /configs/beit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/beit/README.md -------------------------------------------------------------------------------- /configs/beit/beit_base_p16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/beit/beit_base_p16_224.yaml -------------------------------------------------------------------------------- /configs/beit/beit_base_p16_224_ft_1k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/beit/beit_base_p16_224_ft_1k.yaml -------------------------------------------------------------------------------- /configs/beit/beit_base_p16_224_pt_1k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/beit/beit_base_p16_224_pt_1k.yaml -------------------------------------------------------------------------------- /configs/beit/beit_base_p16_384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/beit/beit_base_p16_384.yaml -------------------------------------------------------------------------------- /configs/beit/beit_large_p16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/beit/beit_large_p16_224.yaml -------------------------------------------------------------------------------- /configs/beit/beit_large_p16_384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/beit/beit_large_p16_384.yaml -------------------------------------------------------------------------------- /configs/beit/beit_large_p16_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/beit/beit_large_p16_512.yaml -------------------------------------------------------------------------------- /configs/byol/byol_clas_r50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/byol/byol_clas_r50.yaml -------------------------------------------------------------------------------- /configs/byol/byol_r50_IM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/byol/byol_r50_IM.yaml -------------------------------------------------------------------------------- /configs/cait/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cait/README.md -------------------------------------------------------------------------------- /configs/cait/cait_m36_384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cait/cait_m36_384.yaml -------------------------------------------------------------------------------- /configs/cait/cait_m48_448.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cait/cait_m48_448.yaml -------------------------------------------------------------------------------- /configs/cait/cait_s24_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cait/cait_s24_224.yaml -------------------------------------------------------------------------------- /configs/cait/cait_s24_384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cait/cait_s24_384.yaml -------------------------------------------------------------------------------- /configs/cait/cait_s36_384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cait/cait_s36_384.yaml -------------------------------------------------------------------------------- /configs/cait/cait_xs24_384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cait/cait_xs24_384.yaml -------------------------------------------------------------------------------- /configs/clip/vit-b-32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/clip/vit-b-32.yaml -------------------------------------------------------------------------------- /configs/convnext/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/convnext/README.md -------------------------------------------------------------------------------- /configs/convnext/convnext_small_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/convnext/convnext_small_224.yaml -------------------------------------------------------------------------------- /configs/convnext/convnext_tiny_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/convnext/convnext_tiny_224.yaml -------------------------------------------------------------------------------- /configs/cvt/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cvt/README.md -------------------------------------------------------------------------------- /configs/cvt/cvt_13_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cvt/cvt_13_224.yaml -------------------------------------------------------------------------------- /configs/cvt/cvt_13_384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cvt/cvt_13_384.yaml -------------------------------------------------------------------------------- /configs/cvt/cvt_21_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cvt/cvt_21_224.yaml -------------------------------------------------------------------------------- /configs/cvt/cvt_21_384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cvt/cvt_21_384.yaml -------------------------------------------------------------------------------- /configs/cvt/cvt_w24_384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/cvt/cvt_w24_384.yaml -------------------------------------------------------------------------------- /configs/deit/deit-base-p16-pt_in1k-224_2n16c_fp16_o1_dp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/deit/deit-base-p16-pt_in1k-224_2n16c_fp16_o1_dp.yaml -------------------------------------------------------------------------------- /configs/densecl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/densecl/README.md -------------------------------------------------------------------------------- /configs/densecl/README_ch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/densecl/README_ch.md -------------------------------------------------------------------------------- /configs/densecl/densecl_r50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/densecl/densecl_r50.yaml -------------------------------------------------------------------------------- /configs/lvvit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/lvvit/README.md -------------------------------------------------------------------------------- /configs/lvvit/README_ch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/lvvit/README_ch.md -------------------------------------------------------------------------------- /configs/lvvit/lvvit_tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/lvvit/lvvit_tiny.yaml -------------------------------------------------------------------------------- /configs/mae/mae_vit_b_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/mae/mae_vit_b_finetune.yaml -------------------------------------------------------------------------------- /configs/mae/mae_vit_b_pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/mae/mae_vit_b_pretrain.yaml -------------------------------------------------------------------------------- /configs/mlp_mixer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/mlp_mixer/README.md -------------------------------------------------------------------------------- /configs/mlp_mixer/mlp-mixer_b16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/mlp_mixer/mlp-mixer_b16_224.yaml -------------------------------------------------------------------------------- /configs/mlp_mixer/mlp-mixer_l16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/mlp_mixer/mlp-mixer_l16_224.yaml -------------------------------------------------------------------------------- /configs/moco/moco_clas_r50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/moco/moco_clas_r50.yaml -------------------------------------------------------------------------------- /configs/moco/moco_v1_r50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/moco/moco_v1_r50.yaml -------------------------------------------------------------------------------- /configs/moco/moco_v2_r50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/moco/moco_v2_r50.yaml -------------------------------------------------------------------------------- /configs/moco_byol/moco_byol_r50_IM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/moco_byol/moco_byol_r50_IM.yaml -------------------------------------------------------------------------------- /configs/moco_byol/moco_byol_r50_IM_clas.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/moco_byol/moco_byol_r50_IM_clas.yaml -------------------------------------------------------------------------------- /configs/pixpro/pixpro_base_r50_100ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/pixpro/pixpro_base_r50_100ep.yaml -------------------------------------------------------------------------------- /configs/pixpro/pixpro_base_r50_100ep_IM_clas.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/pixpro/pixpro_base_r50_100ep_IM_clas.yaml -------------------------------------------------------------------------------- /configs/simclr/simclr_r18_cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/simclr/simclr_r18_cifar10.yaml -------------------------------------------------------------------------------- /configs/simclr/simclr_r50_IM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/simclr/simclr_r50_IM.yaml -------------------------------------------------------------------------------- /configs/simsiam/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/simsiam/README.md -------------------------------------------------------------------------------- /configs/simsiam/README_ch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/simsiam/README_ch.md -------------------------------------------------------------------------------- /configs/simsiam/simsiam_clas_r50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/simsiam/simsiam_clas_r50.yaml -------------------------------------------------------------------------------- /configs/simsiam/simsiam_r50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/simsiam/simsiam_r50.yaml -------------------------------------------------------------------------------- /configs/swav/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/swav/README.md -------------------------------------------------------------------------------- /configs/swav/README_ch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/swav/README_ch.md -------------------------------------------------------------------------------- /configs/swav/swav_clas_r50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/swav/swav_clas_r50.yaml -------------------------------------------------------------------------------- /configs/swav/swav_r50_100ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/swav/swav_r50_100ep.yaml -------------------------------------------------------------------------------- /configs/swin_transformer/SwinTransformer_base_patch4_window7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/swin_transformer/SwinTransformer_base_patch4_window7_224.yaml -------------------------------------------------------------------------------- /configs/swin_transformer/SwinTransformer_giant_patch4_window7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/swin_transformer/SwinTransformer_giant_patch4_window7_224.yaml -------------------------------------------------------------------------------- /configs/swin_transformer/SwinTransformer_huge_patch4_window7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/swin_transformer/SwinTransformer_huge_patch4_window7_224.yaml -------------------------------------------------------------------------------- /configs/swin_transformer/SwinTransformer_small_patch4_window7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/swin_transformer/SwinTransformer_small_patch4_window7_224.yaml -------------------------------------------------------------------------------- /configs/swin_transformer/SwinTransformer_tiny_patch4_window7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/swin_transformer/SwinTransformer_tiny_patch4_window7_224.yaml -------------------------------------------------------------------------------- /configs/t2t_vit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/t2t_vit/README.md -------------------------------------------------------------------------------- /configs/t2t_vit/t2t_vit_14.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/t2t_vit/t2t_vit_14.yaml -------------------------------------------------------------------------------- /configs/t2t_vit/t2t_vit_19.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/t2t_vit/t2t_vit_19.yaml -------------------------------------------------------------------------------- /configs/t2t_vit/t2t_vit_24.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/t2t_vit/t2t_vit_24.yaml -------------------------------------------------------------------------------- /configs/t2t_vit/t2t_vit_t_14.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/t2t_vit/t2t_vit_t_14.yaml -------------------------------------------------------------------------------- /configs/t2t_vit/t2t_vit_t_19.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/t2t_vit/t2t_vit_t_19.yaml -------------------------------------------------------------------------------- /configs/t2t_vit/t2t_vit_t_24.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/t2t_vit/t2t_vit_t_24.yaml -------------------------------------------------------------------------------- /configs/vision_transformer/vit-base-p16-ft_in1k-384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/vision_transformer/vit-base-p16-ft_in1k-384.yaml -------------------------------------------------------------------------------- /configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o1_dp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o1_dp.yaml -------------------------------------------------------------------------------- /configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o1_sharding.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o1_sharding.yaml -------------------------------------------------------------------------------- /configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o2_dp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o2_dp.yaml -------------------------------------------------------------------------------- /configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o2_sharding.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o2_sharding.yaml -------------------------------------------------------------------------------- /configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp32_dp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp32_dp.yaml -------------------------------------------------------------------------------- /configs/vision_transformer/vit-base-p32_ft_in1k-384.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/vision_transformer/vit-base-p32_ft_in1k-384.yaml -------------------------------------------------------------------------------- /configs/vision_transformer/vit-g-p14-pt_in1k-224_1n8c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/vision_transformer/vit-g-p14-pt_in1k-224_1n8c.yaml -------------------------------------------------------------------------------- /configs/xcit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/README.md -------------------------------------------------------------------------------- /configs/xcit/xcit_large_24_p16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_large_24_p16_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_large_24_p8_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_large_24_p8_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_large_24_p8_224_dist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_large_24_p8_224_dist.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_medium_24_p16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_medium_24_p16_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_medium_24_p8_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_medium_24_p8_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_medium_24_p8_224_dist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_medium_24_p8_224_dist.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_nano_12_p16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_nano_12_p16_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_nano_12_p8_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_nano_12_p8_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_nano_12_p8_224_dist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_nano_12_p8_224_dist.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_small_12_p16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_small_12_p16_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_small_12_p8_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_small_12_p8_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_small_12_p8_224_dist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_small_12_p8_224_dist.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_small_24_p16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_small_24_p16_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_small_24_p8_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_small_24_p8_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_small_24_p8_224_dist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_small_24_p8_224_dist.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_tiny_12_p16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_tiny_12_p16_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_tiny_12_p8_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_tiny_12_p8_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_tiny_12_p8_224_dist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_tiny_12_p8_224_dist.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_tiny_24_p16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_tiny_24_p16_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_tiny_24_p8_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_tiny_24_p8_224.yaml -------------------------------------------------------------------------------- /configs/xcit/xcit_tiny_24_p8_224_dist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/configs/xcit/xcit_tiny_24_p8_224_dist.yaml -------------------------------------------------------------------------------- /deploy/configs/inference_cls.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/configs/inference_cls.yaml -------------------------------------------------------------------------------- /deploy/images/ILSVRC2012_val_00000010.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/images/ILSVRC2012_val_00000010.jpeg -------------------------------------------------------------------------------- /deploy/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deploy/python/postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/python/postprocess.py -------------------------------------------------------------------------------- /deploy/python/predict_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/python/predict_cls.py -------------------------------------------------------------------------------- /deploy/python/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/python/preprocess.py -------------------------------------------------------------------------------- /deploy/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/utils/__init__.py -------------------------------------------------------------------------------- /deploy/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/utils/config.py -------------------------------------------------------------------------------- /deploy/utils/get_image_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/utils/get_image_list.py -------------------------------------------------------------------------------- /deploy/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/utils/logger.py -------------------------------------------------------------------------------- /deploy/utils/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/deploy/utils/predictor.py -------------------------------------------------------------------------------- /docs/Classification_Models_Guide.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Classification_Models_Guide.md -------------------------------------------------------------------------------- /docs/GETTING_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/GETTING_STARTED.md -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /docs/Large_Scale_Training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Large_Scale_Training.md -------------------------------------------------------------------------------- /docs/Train_BYOL_model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Train_BYOL_model.md -------------------------------------------------------------------------------- /docs/Train_CLIP_model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Train_CLIP_model.md -------------------------------------------------------------------------------- /docs/Train_DenseCL_model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Train_DenseCL_model.md -------------------------------------------------------------------------------- /docs/Train_MoCo-BYOL_model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Train_MoCo-BYOL_model.md -------------------------------------------------------------------------------- /docs/Train_MoCo_model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Train_MoCo_model.md -------------------------------------------------------------------------------- /docs/Train_PixPro_model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Train_PixPro_model.md -------------------------------------------------------------------------------- /docs/Train_SimCLR_model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Train_SimCLR_model.md -------------------------------------------------------------------------------- /docs/Train_SimSiam_model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Train_SimSiam_model.md -------------------------------------------------------------------------------- /docs/Train_SwAV_model.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/Train_SwAV_model.md -------------------------------------------------------------------------------- /docs/imgs/clip_demo1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/imgs/clip_demo1.png -------------------------------------------------------------------------------- /docs/imgs/clip_demo2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/imgs/clip_demo2.png -------------------------------------------------------------------------------- /docs/imgs/densecl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/imgs/densecl.png -------------------------------------------------------------------------------- /docs/imgs/moco-byol.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/imgs/moco-byol.png -------------------------------------------------------------------------------- /docs/imgs/passl_logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/imgs/passl_logo.svg -------------------------------------------------------------------------------- /docs/imgs/simsiam.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/imgs/simsiam.png -------------------------------------------------------------------------------- /docs/imgs/swav.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/docs/imgs/swav.png -------------------------------------------------------------------------------- /passl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/__init__.py -------------------------------------------------------------------------------- /passl/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/core/__init__.py -------------------------------------------------------------------------------- /passl/core/grad_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/core/grad_clip.py -------------------------------------------------------------------------------- /passl/core/grad_scaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/core/grad_scaler.py -------------------------------------------------------------------------------- /passl/core/param_fuse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/core/param_fuse.py -------------------------------------------------------------------------------- /passl/core/recompute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/core/recompute.py -------------------------------------------------------------------------------- /passl/core/sync_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/core/sync_utils.py -------------------------------------------------------------------------------- /passl/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/__init__.py -------------------------------------------------------------------------------- /passl/data/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/dataset/__init__.py -------------------------------------------------------------------------------- /passl/data/dataset/common_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/dataset/common_dataset.py -------------------------------------------------------------------------------- /passl/data/dataset/fewshot_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/dataset/fewshot_dataset.py -------------------------------------------------------------------------------- /passl/data/dataset/imagefolder_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/dataset/imagefolder_dataset.py -------------------------------------------------------------------------------- /passl/data/dataset/imagenet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/dataset/imagenet_dataset.py -------------------------------------------------------------------------------- /passl/data/dataset/swavmulticrop_datatset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/dataset/swavmulticrop_datatset.py -------------------------------------------------------------------------------- /passl/data/postprocess/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/postprocess/__init__.py -------------------------------------------------------------------------------- /passl/data/preprocess/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/preprocess/__init__.py -------------------------------------------------------------------------------- /passl/data/preprocess/basic_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/preprocess/basic_transforms.py -------------------------------------------------------------------------------- /passl/data/preprocess/batch_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/preprocess/batch_transforms.py -------------------------------------------------------------------------------- /passl/data/preprocess/timm_autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/preprocess/timm_autoaugment.py -------------------------------------------------------------------------------- /passl/data/sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/sampler/__init__.py -------------------------------------------------------------------------------- /passl/data/sampler/repeatedaug_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/sampler/repeatedaug_sampler.py -------------------------------------------------------------------------------- /passl/data/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/utils/__init__.py -------------------------------------------------------------------------------- /passl/data/utils/batch_collate_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/utils/batch_collate_fn.py -------------------------------------------------------------------------------- /passl/data/utils/transform_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/data/utils/transform_utils.py -------------------------------------------------------------------------------- /passl/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/distributed/__init__.py -------------------------------------------------------------------------------- /passl/distributed/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/distributed/env.py -------------------------------------------------------------------------------- /passl/distributed/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/distributed/nn/__init__.py -------------------------------------------------------------------------------- /passl/distributed/nn/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/distributed/nn/functional.py -------------------------------------------------------------------------------- /passl/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/engine/__init__.py -------------------------------------------------------------------------------- /passl/engine/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/engine/engine.py -------------------------------------------------------------------------------- /passl/engine/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/engine/inference.py -------------------------------------------------------------------------------- /passl/engine/loops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/engine/loops/__init__.py -------------------------------------------------------------------------------- /passl/engine/loops/classification_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/engine/loops/classification_loop.py -------------------------------------------------------------------------------- /passl/engine/loops/contrastive_learning_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/engine/loops/contrastive_learning_loop.py -------------------------------------------------------------------------------- /passl/engine/loops/loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/engine/loops/loop.py -------------------------------------------------------------------------------- /passl/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/loss/__init__.py -------------------------------------------------------------------------------- /passl/loss/celoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/loss/celoss.py -------------------------------------------------------------------------------- /passl/metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/metric/__init__.py -------------------------------------------------------------------------------- /passl/metric/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/metric/metrics.py -------------------------------------------------------------------------------- /passl/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/__init__.py -------------------------------------------------------------------------------- /passl/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/base_model.py -------------------------------------------------------------------------------- /passl/models/cae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/cae.py -------------------------------------------------------------------------------- /passl/models/cait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/cait.py -------------------------------------------------------------------------------- /passl/models/convmae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/convmae/__init__.py -------------------------------------------------------------------------------- /passl/models/convmae/conv_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/convmae/conv_mae.py -------------------------------------------------------------------------------- /passl/models/convmae/conv_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/convmae/conv_vit.py -------------------------------------------------------------------------------- /passl/models/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/convnext.py -------------------------------------------------------------------------------- /passl/models/deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/deit.py -------------------------------------------------------------------------------- /passl/models/dino/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/dino/__init__.py -------------------------------------------------------------------------------- /passl/models/dino/dino_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/dino/dino_vit.py -------------------------------------------------------------------------------- /passl/models/dinov2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/dinov2/__init__.py -------------------------------------------------------------------------------- /passl/models/dinov2/dinov2_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/dinov2/dinov2_vit.py -------------------------------------------------------------------------------- /passl/models/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/mae.py -------------------------------------------------------------------------------- /passl/models/mocov3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/mocov3.py -------------------------------------------------------------------------------- /passl/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/resnet.py -------------------------------------------------------------------------------- /passl/models/simsiam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/simsiam.py -------------------------------------------------------------------------------- /passl/models/swav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/swav.py -------------------------------------------------------------------------------- /passl/models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/swin_transformer.py -------------------------------------------------------------------------------- /passl/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/utils/__init__.py -------------------------------------------------------------------------------- /passl/models/utils/averaged_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/utils/averaged_model.py -------------------------------------------------------------------------------- /passl/models/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/utils/ema.py -------------------------------------------------------------------------------- /passl/models/utils/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/utils/pos_embed.py -------------------------------------------------------------------------------- /passl/models/utils/tome.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/utils/tome.py -------------------------------------------------------------------------------- /passl/models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/vision_transformer.py -------------------------------------------------------------------------------- /passl/models/vision_transformer_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/models/vision_transformer_hybrid.py -------------------------------------------------------------------------------- /passl/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/nn/__init__.py -------------------------------------------------------------------------------- /passl/nn/finer_grained_mp_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/nn/finer_grained_mp_layers.py -------------------------------------------------------------------------------- /passl/nn/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/nn/init.py -------------------------------------------------------------------------------- /passl/nn/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/nn/norm.py -------------------------------------------------------------------------------- /passl/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/__init__.py -------------------------------------------------------------------------------- /passl/optimizer/adafactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/adafactor.py -------------------------------------------------------------------------------- /passl/optimizer/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/adamw.py -------------------------------------------------------------------------------- /passl/optimizer/adan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/adan.py -------------------------------------------------------------------------------- /passl/optimizer/momentum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/momentum.py -------------------------------------------------------------------------------- /passl/optimizer/momentum_larc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/momentum_larc.py -------------------------------------------------------------------------------- /passl/optimizer/momentum_lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/momentum_lars.py -------------------------------------------------------------------------------- /passl/optimizer/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/optimizer.py -------------------------------------------------------------------------------- /passl/optimizer/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/utils/__init__.py -------------------------------------------------------------------------------- /passl/optimizer/utils/group_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/optimizer/utils/group_params.py -------------------------------------------------------------------------------- /passl/scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/scheduler/__init__.py -------------------------------------------------------------------------------- /passl/scheduler/lr_callable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/scheduler/lr_callable.py -------------------------------------------------------------------------------- /passl/scheduler/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/scheduler/lr_scheduler.py -------------------------------------------------------------------------------- /passl/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/utils/__init__.py -------------------------------------------------------------------------------- /passl/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/utils/config.py -------------------------------------------------------------------------------- /passl/utils/infohub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/utils/infohub.py -------------------------------------------------------------------------------- /passl/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/utils/io.py -------------------------------------------------------------------------------- /passl/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/utils/logger.py -------------------------------------------------------------------------------- /passl/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/utils/misc.py -------------------------------------------------------------------------------- /passl/utils/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl/utils/profiler.py -------------------------------------------------------------------------------- /passl_v110/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/__init__.py -------------------------------------------------------------------------------- /passl_v110/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/__init__.py -------------------------------------------------------------------------------- /passl_v110/datasets/beitdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/beitdataset.py -------------------------------------------------------------------------------- /passl_v110/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/builder.py -------------------------------------------------------------------------------- /passl_v110/datasets/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/cifar.py -------------------------------------------------------------------------------- /passl_v110/datasets/folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/folder.py -------------------------------------------------------------------------------- /passl_v110/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/imagenet.py -------------------------------------------------------------------------------- /passl_v110/datasets/lvdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/lvdataset.py -------------------------------------------------------------------------------- /passl_v110/datasets/multicropdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/multicropdataset.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/__init__.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/auto_augment.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/builder.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/constants.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/cv2_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/cv2_func.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/cv2_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/cv2_trans.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/img_pil_pixpro_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/img_pil_pixpro_transforms.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/lvvit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/lvvit/__init__.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/lvvit/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/lvvit/auto_augment.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/lvvit/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/lvvit/mixup.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/lvvit/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/lvvit/transforms.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/masking_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/masking_generator.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/mixup.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/random_erasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/random_erasing.py -------------------------------------------------------------------------------- /passl_v110/datasets/preprocess/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/preprocess/transforms.py -------------------------------------------------------------------------------- /passl_v110/datasets/textimagedataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/datasets/textimagedataset.py -------------------------------------------------------------------------------- /passl_v110/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/engine/__init__.py -------------------------------------------------------------------------------- /passl_v110/engine/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/engine/trainer.py -------------------------------------------------------------------------------- /passl_v110/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/__init__.py -------------------------------------------------------------------------------- /passl_v110/hooks/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/builder.py -------------------------------------------------------------------------------- /passl_v110/hooks/checkpoint_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/checkpoint_hook.py -------------------------------------------------------------------------------- /passl_v110/hooks/evaluate_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/evaluate_hook.py -------------------------------------------------------------------------------- /passl_v110/hooks/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/hook.py -------------------------------------------------------------------------------- /passl_v110/hooks/log_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/log_hook.py -------------------------------------------------------------------------------- /passl_v110/hooks/lr_scheduler_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/lr_scheduler_hook.py -------------------------------------------------------------------------------- /passl_v110/hooks/optimizer_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/optimizer_hook.py -------------------------------------------------------------------------------- /passl_v110/hooks/timer_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/timer_hook.py -------------------------------------------------------------------------------- /passl_v110/hooks/visual_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/hooks/visual_hook.py -------------------------------------------------------------------------------- /passl_v110/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/BEiTWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/BEiTWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/BYOL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/BYOL.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/CLIPWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/CLIPWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/CaiTWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/CaiTWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/CvTWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/CvTWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/DeiTWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/DeiTWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/DistillationWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/DistillationWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/LVViTWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/LVViTWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/MAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/MAE.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/MlpMixerWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/MlpMixerWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/MoCoBYOL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/MoCoBYOL.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/SwinWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/SwinWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/T2TViTWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/T2TViTWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/ViTWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/ViTWrapper.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/__init__.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/builder.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/byol_clas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/byol_clas.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/clas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/clas.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/densecl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/densecl.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/moco.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/pixpro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/pixpro.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/simclr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/simclr.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/simsiam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/simsiam.py -------------------------------------------------------------------------------- /passl_v110/modeling/architectures/swav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/architectures/swav.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/__init__.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/base_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/base_transformer.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/beit.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/beit_ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/beit_ft.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/beit_pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/beit_pt.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/builder.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/cait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/cait.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/clip.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/convnext.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/cvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/cvt.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/deit.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/discrete_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/discrete_vae.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/lvvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/lvvit.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/mae.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/mlp_mixer.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/regnet.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/resnet.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/resnetcifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/resnetcifar.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/resnetimagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/resnetimagenet.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/resnetsimclr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/resnetsimclr.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/resnetswav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/resnetswav.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/swin_transformer.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/t2t_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/t2t_vit.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/transformer.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/vision_transformer.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/vit.py -------------------------------------------------------------------------------- /passl_v110/modeling/backbones/xcit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/backbones/xcit.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/__init__.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/beit_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/beit_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/builder.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/cait_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/cait_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/clas_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/clas_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/clip_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/clip_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/contrastive_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/contrastive_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/cvt_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/cvt_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/l2_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/l2_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/mb_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/mb_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/mlp_mixer_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/mlp_mixer_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/pixpro_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/pixpro_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/simclr_contrastive_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/simclr_contrastive_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/simsiam_contrastive_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/simsiam_contrastive_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/swav_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/swav_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/swin_transformer_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/swin_transformer_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/t2t_vit_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/t2t_vit_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/heads/vision_transformer_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/heads/vision_transformer_head.py -------------------------------------------------------------------------------- /passl_v110/modeling/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/necks/__init__.py -------------------------------------------------------------------------------- /passl_v110/modeling/necks/base_neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/necks/base_neck.py -------------------------------------------------------------------------------- /passl_v110/modeling/necks/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modeling/necks/builder.py -------------------------------------------------------------------------------- /passl_v110/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modules/__init__.py -------------------------------------------------------------------------------- /passl_v110/modules/freeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modules/freeze.py -------------------------------------------------------------------------------- /passl_v110/modules/get_sincos_pe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modules/get_sincos_pe.py -------------------------------------------------------------------------------- /passl_v110/modules/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/modules/init.py -------------------------------------------------------------------------------- /passl_v110/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/solver/__init__.py -------------------------------------------------------------------------------- /passl_v110/solver/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/solver/builder.py -------------------------------------------------------------------------------- /passl_v110/solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/solver/lr_scheduler.py -------------------------------------------------------------------------------- /passl_v110/solver/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/solver/optimizer.py -------------------------------------------------------------------------------- /passl_v110/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/__init__.py -------------------------------------------------------------------------------- /passl_v110/utils/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /passl_v110/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/config.py -------------------------------------------------------------------------------- /passl_v110/utils/imagenet1k_label_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/imagenet1k_label_list.txt -------------------------------------------------------------------------------- /passl_v110/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/logger.py -------------------------------------------------------------------------------- /passl_v110/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/misc.py -------------------------------------------------------------------------------- /passl_v110/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/options.py -------------------------------------------------------------------------------- /passl_v110/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/registry.py -------------------------------------------------------------------------------- /passl_v110/utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/setup.py -------------------------------------------------------------------------------- /passl_v110/utils/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/passl_v110/utils/tokenizer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/setup.py -------------------------------------------------------------------------------- /tasks/accelerate/tome/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/accelerate/tome/README.md -------------------------------------------------------------------------------- /tasks/accelerate/tome/validation_tome_vit.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/accelerate/tome/validation_tome_vit.ipynb -------------------------------------------------------------------------------- /tasks/classification/cait/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/cait/README.md -------------------------------------------------------------------------------- /tasks/classification/cait/configs/cait_s24_224_in1k_1n8c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/cait/configs/cait_s24_224_in1k_1n8c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/cait/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/cait/train.sh -------------------------------------------------------------------------------- /tasks/classification/convnext/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/convnext/README.md -------------------------------------------------------------------------------- /tasks/classification/convnext/configs/ConvNeXt_base_224_in1k_1n8c_dp_fp32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/convnext/configs/ConvNeXt_base_224_in1k_1n8c_dp_fp32.yaml -------------------------------------------------------------------------------- /tasks/classification/convnext/configs/ConvNeXt_base_224_in1k_4n32c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/convnext/configs/ConvNeXt_base_224_in1k_4n32c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/convnext/configs/ConvNeXt_base_224_in1k_4n32c_dp_fp32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/convnext/configs/ConvNeXt_base_224_in1k_4n32c_dp_fp32.yaml -------------------------------------------------------------------------------- /tasks/classification/deit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/deit/README.md -------------------------------------------------------------------------------- /tasks/classification/deit/configs/DeiT_base_patch16_224_in1k_1n8c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/deit/configs/DeiT_base_patch16_224_in1k_1n8c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/deit/configs/DeiT_base_patch16_224_in1k_1n8c_dp_fp32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/deit/configs/DeiT_base_patch16_224_in1k_1n8c_dp_fp32.yaml -------------------------------------------------------------------------------- /tasks/classification/deit/configs/DeiT_base_patch16_224_in1k_2n16c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/deit/configs/DeiT_base_patch16_224_in1k_2n16c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/swin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/swin/README.md -------------------------------------------------------------------------------- /tasks/classification/swin/configs/swin_base_patch4_window7_224_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/swin/configs/swin_base_patch4_window7_224_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/classification/swin/configs/swin_base_patch4_window7_224_in1k_1n8c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/swin/configs/swin_base_patch4_window7_224_in1k_1n8c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/swin/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/swin/train.sh -------------------------------------------------------------------------------- /tasks/classification/vit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/vit/README.md -------------------------------------------------------------------------------- /tasks/classification/vit/configs/ViT_base_patch16_224_in1k_1n8c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/vit/configs/ViT_base_patch16_224_in1k_1n8c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/vit/configs/ViT_base_patch16_224_in1k_adan_1n8c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/vit/configs/ViT_base_patch16_224_in1k_adan_1n8c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/vit/configs/ViT_base_patch16_384_ft_in1k_1n8c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/vit/configs/ViT_base_patch16_384_ft_in1k_1n8c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/vit/configs/ViT_large_patch16_224_in21k_4n32c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/vit/configs/ViT_large_patch16_224_in21k_4n32c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/vit/configs/ViT_large_patch16_384_in1k_ft_4n32c_dp_fp16o2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/vit/configs/ViT_large_patch16_384_in1k_ft_4n32c_dp_fp16o2.yaml -------------------------------------------------------------------------------- /tasks/classification/vit/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/vit/eval.sh -------------------------------------------------------------------------------- /tasks/classification/vit/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/classification/vit/evaluate.py -------------------------------------------------------------------------------- /tasks/ssl/cae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/README.md -------------------------------------------------------------------------------- /tasks/ssl/cae/dall_e/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/dall_e/__init__.py -------------------------------------------------------------------------------- /tasks/ssl/cae/dall_e/dalle_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/dall_e/dalle_vae.py -------------------------------------------------------------------------------- /tasks/ssl/cae/dall_e/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/dall_e/decoder.py -------------------------------------------------------------------------------- /tasks/ssl/cae/dall_e/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/dall_e/encoder.py -------------------------------------------------------------------------------- /tasks/ssl/cae/dall_e/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/dall_e/utils.py -------------------------------------------------------------------------------- /tasks/ssl/cae/engine_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/engine_finetune.py -------------------------------------------------------------------------------- /tasks/ssl/cae/engine_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/engine_pretrain.py -------------------------------------------------------------------------------- /tasks/ssl/cae/extract_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/extract_model.py -------------------------------------------------------------------------------- /tasks/ssl/cae/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/finetune.sh -------------------------------------------------------------------------------- /tasks/ssl/cae/linprobe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/linprobe.sh -------------------------------------------------------------------------------- /tasks/ssl/cae/main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/main_finetune.py -------------------------------------------------------------------------------- /tasks/ssl/cae/main_linprobe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/main_linprobe.py -------------------------------------------------------------------------------- /tasks/ssl/cae/main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/main_pretrain.py -------------------------------------------------------------------------------- /tasks/ssl/cae/pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/pretrain.sh -------------------------------------------------------------------------------- /tasks/ssl/cae/util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/util/loss.py -------------------------------------------------------------------------------- /tasks/ssl/cae/util/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/util/lr_decay.py -------------------------------------------------------------------------------- /tasks/ssl/cae/util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/util/lr_sched.py -------------------------------------------------------------------------------- /tasks/ssl/cae/util/masking_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/util/masking_generator.py -------------------------------------------------------------------------------- /tasks/ssl/cae/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/cae/util/misc.py -------------------------------------------------------------------------------- /tasks/ssl/dino/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dino/README.md -------------------------------------------------------------------------------- /tasks/ssl/dino/configs/dino_deit_small_patch16_224_lp_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dino/configs/dino_deit_small_patch16_224_lp_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/dino/configs/dino_deit_small_patch8_224_lp_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dino/configs/dino_deit_small_patch8_224_lp_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/dino/configs/dino_vit_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dino/configs/dino_vit_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/dino/configs/dino_vit_base_patch8_224_lp_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dino/configs/dino_vit_base_patch8_224_lp_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/dino/linearprobe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dino/linearprobe.sh -------------------------------------------------------------------------------- /tasks/ssl/dinov2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dinov2/README.md -------------------------------------------------------------------------------- /tasks/ssl/dinov2/configs/dinov2_vit_base_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dinov2/configs/dinov2_vit_base_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/dinov2/configs/dinov2_vit_gaint2_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dinov2/configs/dinov2_vit_gaint2_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/dinov2/configs/dinov2_vit_large_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dinov2/configs/dinov2_vit_large_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/dinov2/configs/dinov2_vit_small_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dinov2/configs/dinov2_vit_small_patch14_224_lp_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/dinov2/linearprobe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/dinov2/linearprobe.sh -------------------------------------------------------------------------------- /tasks/ssl/mae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/README.md -------------------------------------------------------------------------------- /tasks/ssl/mae/engine_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/engine_finetune.py -------------------------------------------------------------------------------- /tasks/ssl/mae/engine_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/engine_pretrain.py -------------------------------------------------------------------------------- /tasks/ssl/mae/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/finetune.sh -------------------------------------------------------------------------------- /tasks/ssl/mae/finetune_convmae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/finetune_convmae.sh -------------------------------------------------------------------------------- /tasks/ssl/mae/linprobe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/linprobe.sh -------------------------------------------------------------------------------- /tasks/ssl/mae/linprobe_convmae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/linprobe_convmae.sh -------------------------------------------------------------------------------- /tasks/ssl/mae/main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/main_finetune.py -------------------------------------------------------------------------------- /tasks/ssl/mae/main_linprobe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/main_linprobe.py -------------------------------------------------------------------------------- /tasks/ssl/mae/main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/main_pretrain.py -------------------------------------------------------------------------------- /tasks/ssl/mae/pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/pretrain.sh -------------------------------------------------------------------------------- /tasks/ssl/mae/pretrain_convmae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/pretrain_convmae.sh -------------------------------------------------------------------------------- /tasks/ssl/mae/util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/util/loss.py -------------------------------------------------------------------------------- /tasks/ssl/mae/util/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/util/lr_decay.py -------------------------------------------------------------------------------- /tasks/ssl/mae/util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/util/lr_sched.py -------------------------------------------------------------------------------- /tasks/ssl/mae/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/util/misc.py -------------------------------------------------------------------------------- /tasks/ssl/mae/util/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/util/optim_factory.py -------------------------------------------------------------------------------- /tasks/ssl/mae/util/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mae/util/pos_embed.py -------------------------------------------------------------------------------- /tasks/ssl/mocov3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/README.md -------------------------------------------------------------------------------- /tasks/ssl/mocov3/builder_moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/builder_moco.py -------------------------------------------------------------------------------- /tasks/ssl/mocov3/configs/mocov3_deit_base_patch16_224_ft_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/configs/mocov3_deit_base_patch16_224_ft_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/mocov3/configs/mocov3_vit_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/configs/mocov3_vit_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/mocov3/configs/mocov3_vit_base_patch16_224_pt_in1k_4n32c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/configs/mocov3_vit_base_patch16_224_pt_in1k_4n32c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/mocov3/extract_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/extract_weight.py -------------------------------------------------------------------------------- /tasks/ssl/mocov3/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/finetune.sh -------------------------------------------------------------------------------- /tasks/ssl/mocov3/linearprobe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/linearprobe.sh -------------------------------------------------------------------------------- /tasks/ssl/mocov3/main_lincls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/main_lincls.py -------------------------------------------------------------------------------- /tasks/ssl/mocov3/main_moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/main_moco.py -------------------------------------------------------------------------------- /tasks/ssl/mocov3/pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/pretrain.sh -------------------------------------------------------------------------------- /tasks/ssl/mocov3/vit_moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/mocov3/vit_moco.py -------------------------------------------------------------------------------- /tasks/ssl/simsiam/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/simsiam/README.md -------------------------------------------------------------------------------- /tasks/ssl/simsiam/configs/simsiam_resnet50_lp_in1k_1n8c_dp_fp32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/simsiam/configs/simsiam_resnet50_lp_in1k_1n8c_dp_fp32.yaml -------------------------------------------------------------------------------- /tasks/ssl/simsiam/configs/simsiam_resnet50_pt_in1k_1n8c_dp_fp32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/simsiam/configs/simsiam_resnet50_pt_in1k_1n8c_dp_fp32.yaml -------------------------------------------------------------------------------- /tasks/ssl/simsiam/linearprobe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/simsiam/linearprobe.sh -------------------------------------------------------------------------------- /tasks/ssl/simsiam/pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/simsiam/pretrain.sh -------------------------------------------------------------------------------- /tasks/ssl/swav/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/swav/README.md -------------------------------------------------------------------------------- /tasks/ssl/swav/configs/swav_resnet50_224_ft_in1k_1n4c_dp_fp32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/swav/configs/swav_resnet50_224_ft_in1k_1n4c_dp_fp32.yaml -------------------------------------------------------------------------------- /tasks/ssl/swav/configs/swav_resnet50_224_lp_in1k_1n8c_dp_fp32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/swav/configs/swav_resnet50_224_lp_in1k_1n8c_dp_fp32.yaml -------------------------------------------------------------------------------- /tasks/ssl/swav/configs/swav_resnet50_224_pt_in1k_1n8c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/swav/configs/swav_resnet50_224_pt_in1k_1n8c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/swav/configs/swav_resnet50_224_pt_in1k_4n32c_dp_fp16o1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/swav/configs/swav_resnet50_224_pt_in1k_4n32c_dp_fp16o1.yaml -------------------------------------------------------------------------------- /tasks/ssl/swav/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/swav/finetune.sh -------------------------------------------------------------------------------- /tasks/ssl/swav/linearprobe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/swav/linearprobe.sh -------------------------------------------------------------------------------- /tasks/ssl/swav/pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/swav/pretrain.sh -------------------------------------------------------------------------------- /tasks/ssl/swav/pretrain_1N8C.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tasks/ssl/swav/pretrain_1N8C.sh -------------------------------------------------------------------------------- /tests/CI/before_hook.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/before_hook.sh -------------------------------------------------------------------------------- /tests/CI/case.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/case.sh -------------------------------------------------------------------------------- /tests/CI/classification/cait/cait_s24_224_in1k_1n8c_dp_fp16o2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/classification/cait/cait_s24_224_in1k_1n8c_dp_fp16o2.sh -------------------------------------------------------------------------------- /tests/CI/classification/convnext/ConvNeXt_base_224_in1k_1n8c_dp_fp32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/classification/convnext/ConvNeXt_base_224_in1k_1n8c_dp_fp32.sh -------------------------------------------------------------------------------- /tests/CI/classification/deit/DeiT_base_patch16_224_in1k_1n8c_dp_fp16o2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/classification/deit/DeiT_base_patch16_224_in1k_1n8c_dp_fp16o2.sh -------------------------------------------------------------------------------- /tests/CI/classification/deit/DeiT_base_patch16_224_in1k_1n8c_dp_fp32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/classification/deit/DeiT_base_patch16_224_in1k_1n8c_dp_fp32.sh -------------------------------------------------------------------------------- /tests/CI/classification/swin/swin_base_patch4_window7_224_fp16o2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/classification/swin/swin_base_patch4_window7_224_fp16o2.sh -------------------------------------------------------------------------------- /tests/CI/classification/vit/ViT_base_patch16_224_in1k_1n8c_dp_fp16o2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/classification/vit/ViT_base_patch16_224_in1k_1n8c_dp_fp16o2.sh -------------------------------------------------------------------------------- /tests/CI/classification/vit/ViT_base_patch16_384_ft_in1k_1n8c_dp_fp16o2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/classification/vit/ViT_base_patch16_384_ft_in1k_1n8c_dp_fp16o2.sh -------------------------------------------------------------------------------- /tests/CI/end_hook.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/end_hook.sh -------------------------------------------------------------------------------- /tests/CI/run_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/run_all.sh -------------------------------------------------------------------------------- /tests/CI/ssl/cae/cae_base_patch16_224_ft_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/cae/cae_base_patch16_224_ft_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/cae/cae_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/cae/cae_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/cae/cae_base_patch16_224_pt_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/cae/cae_base_patch16_224_pt_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/convmae/convmae_convvit_base_patch16_ft_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/convmae/convmae_convvit_base_patch16_ft_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/convmae/convmae_convvit_base_patch16_lp_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/convmae/convmae_convvit_base_patch16_lp_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/convmae/convmae_convvit_base_patch16_pt_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/convmae/convmae_convvit_base_patch16_pt_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/dino/dino_deit_small_patch16_224_lp_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/dino/dino_deit_small_patch16_224_lp_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/dinov2/dinov2_vit_small_patch14_224_lp_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/dinov2/dinov2_vit_small_patch14_224_lp_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/mae/mae_vit_base_patch16_ft_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/mae/mae_vit_base_patch16_ft_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/mae/mae_vit_base_patch16_lp_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/mae/mae_vit_base_patch16_lp_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/mae/mae_vit_base_patch16_pt_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/mae/mae_vit_base_patch16_pt_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/mocov3/mocov3_deit_base_patch16_224_ft_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/mocov3/mocov3_deit_base_patch16_224_ft_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/mocov3/mocov3_vit_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/mocov3/mocov3_vit_base_patch16_224_lp_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/mocov3/mocov3_vit_base_patch16_224_pt_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/mocov3/mocov3_vit_base_patch16_224_pt_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/CI/ssl/simsiam/simsiam_resnet50_lp_in1k_1n8c_dp_fp32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/simsiam/simsiam_resnet50_lp_in1k_1n8c_dp_fp32.sh -------------------------------------------------------------------------------- /tests/CI/ssl/simsiam/simsiam_resnet50_pt_in1k_1n8c_dp_fp32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/simsiam/simsiam_resnet50_pt_in1k_1n8c_dp_fp32.sh -------------------------------------------------------------------------------- /tests/CI/ssl/swav/swav_resnet50_224_ft_in1k_1n4c_dp_fp32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/swav/swav_resnet50_224_ft_in1k_1n4c_dp_fp32.sh -------------------------------------------------------------------------------- /tests/CI/ssl/swav/swav_resnet50_224_lp_in1k_1n8c_dp_fp32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/swav/swav_resnet50_224_lp_in1k_1n8c_dp_fp32.sh -------------------------------------------------------------------------------- /tests/CI/ssl/swav/swav_resnet50_224_pt_in1k_1n8c_dp_fp16o1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/CI/ssl/swav/swav_resnet50_224_pt_in1k_1n8c_dp_fp16o1.sh -------------------------------------------------------------------------------- /tests/test_tipc/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_tipc/classification/N1C8/ConvNeXt_base_224_bs512_fp32_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/classification/N1C8/ConvNeXt_base_224_bs512_fp32_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/classification/N1C8/DeiT_base_patch16_224_bs128_fp16o2_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/classification/N1C8/DeiT_base_patch16_224_bs128_fp16o2_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/classification/N1C8/DeiT_base_patch16_224_bs128_fp32_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/classification/N1C8/DeiT_base_patch16_224_bs128_fp32_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/classification/N1C8/ViT_base_patch16_224_bs512_fp16_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/classification/N1C8/ViT_base_patch16_224_bs512_fp16_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/classification/N1C8/ViT_base_patch16_384_ft_bs512_fp16_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/classification/N1C8/ViT_base_patch16_384_ft_bs512_fp16_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/classification/N1C8/cait_s24_224_bs128_fp16o2_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/classification/N1C8/cait_s24_224_bs128_fp16o2_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/classification/N1C8/swin_base_patch4_window7_224_bs128_fp16o2_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/classification/N1C8/swin_base_patch4_window7_224_bs128_fp16o2_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/classification/benchmark_common/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/classification/benchmark_common/prepare.sh -------------------------------------------------------------------------------- /tests/test_tipc/classification/benchmark_common/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/classification/benchmark_common/run_benchmark.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/cae/N1C8/cae_base_patch16_224_ft_bs128_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/cae/N1C8/cae_base_patch16_224_ft_bs128_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/cae/N1C8/cae_base_patch16_224_lp_bs512_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/cae/N1C8/cae_base_patch16_224_lp_bs512_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/cae/N1C8/cae_base_patch16_224_pt_bs64_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/cae/N1C8/cae_base_patch16_224_pt_bs64_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/cae/benchmark_common/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/cae/benchmark_common/prepare.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/cae/benchmark_common/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/cae/benchmark_common/run_benchmark.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/convmae/N1C8/convmae_convvit_base_patch16_ft_bs32_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/convmae/N1C8/convmae_convvit_base_patch16_ft_bs32_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/convmae/N1C8/convmae_convvit_base_patch16_lp_bs128_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/convmae/N1C8/convmae_convvit_base_patch16_lp_bs128_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/convmae/N1C8/convmae_convvit_base_patch16_pt_bs64_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/convmae/N1C8/convmae_convvit_base_patch16_pt_bs64_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/convmae/benchmark_common/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/convmae/benchmark_common/prepare.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/convmae/benchmark_common/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/convmae/benchmark_common/run_benchmark.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mae/N1C8/mae_vit_base_patch16_ft_bs32_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mae/N1C8/mae_vit_base_patch16_ft_bs32_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mae/N1C8/mae_vit_base_patch16_lp_bs512_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mae/N1C8/mae_vit_base_patch16_lp_bs512_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mae/N1C8/mae_vit_base_patch16_pt_bs128_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mae/N1C8/mae_vit_base_patch16_pt_bs128_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mae/benchmark_common/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mae/benchmark_common/prepare.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mae/benchmark_common/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mae/benchmark_common/run_benchmark.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mocov3/N1C8/mocov3_deit_base_patch16_224_ft_bs128_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mocov3/N1C8/mocov3_deit_base_patch16_224_ft_bs128_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mocov3/N1C8/mocov3_vit_base_patch16_224_lp_bs128_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mocov3/N1C8/mocov3_vit_base_patch16_224_lp_bs128_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mocov3/N1C8/mocov3_vit_base_patch16_224_pt_bs64_fp16o1_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mocov3/N1C8/mocov3_vit_base_patch16_224_pt_bs64_fp16o1_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mocov3/benchmark_common/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mocov3/benchmark_common/prepare.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/mocov3/benchmark_common/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/mocov3/benchmark_common/run_benchmark.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/simsiam/N1C8/simsiam_resnet50_lp_bs512_fp32_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/simsiam/N1C8/simsiam_resnet50_lp_bs512_fp32_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/simsiam/N1C8/simsiam_resnet50_pt_bs64_fp32_DP8-MP1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/simsiam/N1C8/simsiam_resnet50_pt_bs64_fp32_DP8-MP1.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/simsiam/benchmark_common/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/simsiam/benchmark_common/prepare.sh -------------------------------------------------------------------------------- /tests/test_tipc/ssl/simsiam/benchmark_common/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc/ssl/simsiam/benchmark_common/run_benchmark.sh -------------------------------------------------------------------------------- /tests/test_tipc_v110/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/README.md -------------------------------------------------------------------------------- /tests/test_tipc_v110/common_func.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/common_func.sh -------------------------------------------------------------------------------- /tests/test_tipc_v110/config/densecl_r50/train_infer_python.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/config/densecl_r50/train_infer_python.txt -------------------------------------------------------------------------------- /tests/test_tipc_v110/config/lvvit/train_infer_python.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/config/lvvit/train_infer_python.txt -------------------------------------------------------------------------------- /tests/test_tipc_v110/config/moco_v1_r50/train_infer_python.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/config/moco_v1_r50/train_infer_python.txt -------------------------------------------------------------------------------- /tests/test_tipc_v110/config/simsiam_r50/train_infer_python.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/config/simsiam_r50/train_infer_python.txt -------------------------------------------------------------------------------- /tests/test_tipc_v110/config/swav_r50_100ep/train_infer_python.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/config/swav_r50_100ep/train_infer_python.txt -------------------------------------------------------------------------------- /tests/test_tipc_v110/docs/guide.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/docs/guide.png -------------------------------------------------------------------------------- /tests/test_tipc_v110/docs/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/docs/install.md -------------------------------------------------------------------------------- /tests/test_tipc_v110/docs/test_train_inference_python.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/docs/test_train_inference_python.md -------------------------------------------------------------------------------- /tests/test_tipc_v110/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/prepare.sh -------------------------------------------------------------------------------- /tests/test_tipc_v110/test_train_inference_python.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tests/test_tipc_v110/test_train_inference_python.sh -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools/export.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools_v110/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools_v110/__init__.py -------------------------------------------------------------------------------- /tools_v110/extract_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools_v110/extract_weight.py -------------------------------------------------------------------------------- /tools_v110/model_stat/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools_v110/model_stat/__init__.py -------------------------------------------------------------------------------- /tools_v110/model_stat/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools_v110/model_stat/get_flops.py -------------------------------------------------------------------------------- /tools_v110/model_stat/get_throughput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools_v110/model_stat/get_throughput.py -------------------------------------------------------------------------------- /tools_v110/model_stat/op_count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools_v110/model_stat/op_count.py -------------------------------------------------------------------------------- /tools_v110/passl2ppclas/convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools_v110/passl2ppclas/convert.py -------------------------------------------------------------------------------- /tools_v110/passl2ppclas/ppclas_res50_keys.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools_v110/passl2ppclas/ppclas_res50_keys.txt -------------------------------------------------------------------------------- /tools_v110/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/tools_v110/train.py -------------------------------------------------------------------------------- /version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PaddlePaddle/PASSL/HEAD/version.py --------------------------------------------------------------------------------