├── .gitignore ├── LICENSE ├── __init__.py ├── prototype ├── __init__.py ├── data │ ├── __init__.py │ ├── auto_augmentation.py │ ├── composition.py │ ├── custom_dataloader.py │ ├── datasets │ │ ├── __init__.py │ │ ├── base_dataset.py │ │ ├── custom_dataset.py │ │ ├── imagenet_dataset.py │ │ ├── imagnetc.py │ │ └── imagnetp.py │ ├── image_reader.py │ ├── imagenet_dataloader.py │ ├── metrics │ │ ├── __init__.py │ │ ├── base_evaluator.py │ │ ├── custom_evaluator.py │ │ ├── imagenet_evaluator.py │ │ ├── imagenetc_evaluator.py │ │ └── multiclass_evaluator.py │ ├── nvidia_dali_dataloader.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── imagenet_pipeline.py │ │ └── imagenet_pipeline_v2.py │ ├── register_factory.py │ ├── sampler.py │ ├── transforms.py │ └── utils │ │ ├── __init__.py │ │ ├── calibration_tools.py │ │ └── video_loader.py ├── loss_functions │ ├── __init__.py │ ├── loss.py │ └── nt_xent.py ├── lr_scheduler │ ├── __init__.py │ └── scheduler.py ├── model │ ├── __init__.py │ ├── alexnet.py │ ├── densenet.py │ ├── efficientnet.py │ ├── layer │ │ ├── __init__.py │ │ ├── condconv.py │ │ ├── drop_path.py │ │ ├── sparse_layer.py │ │ ├── splat.py │ │ └── weightnet.py │ ├── mobilenet_v2.py │ ├── mobilenet_v3.py │ ├── regnet.py │ ├── repvgg.py │ ├── resnet_official.py │ ├── shufflenet_v2.py │ ├── vision_transformer.py │ └── vit │ │ ├── __init__.py │ │ ├── default_cfg.py │ │ ├── layers │ │ ├── __init__.py │ │ ├── attention │ │ │ ├── __init__.py │ │ │ ├── levit_attention.py │ │ │ ├── swin_attention.py │ │ │ ├── token2token_attention.py │ │ │ └── vanilla_msa.py │ │ ├── drop_path.py │ │ ├── helpers.py │ │ ├── mlp │ │ │ ├── __init__.py │ │ │ └── vanilla_mlp.py │ │ ├── patch_embedding │ │ │ ├── __init__.py │ │ │ ├── levit_hybrid_backbone.py │ │ │ ├── token2token_embedding.py │ │ │ └── vanilla_embedding.py │ │ └── position_embedding.py │ │ ├── mlp_mixer.py │ │ ├── swin_loader.py │ │ ├── swin_transformer.py │ │ ├── token2token_vit.py │ │ └── vit_base.py ├── optimizer │ └── __init__.py ├── solver │ ├── README.md │ ├── __init__.py │ ├── adv_cls_solver_train_pgd.py │ ├── base_solver.py │ ├── benchmark_eval_adv.py │ ├── cls_solver.py │ ├── cls_solver_robust.py │ ├── imgnet_a_o_eval_solver.py │ ├── imgnet_c_solver.py │ ├── imgnet_p_eval_solver.py │ ├── mimic_eval_solver.py │ ├── multi_eval_decoder_resize_solver.py │ ├── reparam_solver.py │ └── vit_solver.py ├── tools │ ├── Draw_resize_diff.ipynb │ ├── __init__.py │ └── inference.py └── utils │ ├── __init__.py │ ├── dist.py │ ├── ema.py │ ├── misc.py │ ├── model_config.py │ ├── register.py │ └── trunc_normal_initializer.py ├── requirements.txt └── spring ├── __init__.py └── linklink ├── __init__.py └── nn.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/LICENSE -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /prototype/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = 'v3.0.0-beta' 2 | -------------------------------------------------------------------------------- /prototype/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/__init__.py -------------------------------------------------------------------------------- /prototype/data/auto_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/auto_augmentation.py -------------------------------------------------------------------------------- /prototype/data/composition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/composition.py -------------------------------------------------------------------------------- /prototype/data/custom_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/custom_dataloader.py -------------------------------------------------------------------------------- /prototype/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/datasets/__init__.py -------------------------------------------------------------------------------- /prototype/data/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/datasets/base_dataset.py -------------------------------------------------------------------------------- /prototype/data/datasets/custom_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/datasets/custom_dataset.py -------------------------------------------------------------------------------- /prototype/data/datasets/imagenet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/datasets/imagenet_dataset.py -------------------------------------------------------------------------------- /prototype/data/datasets/imagnetc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/datasets/imagnetc.py -------------------------------------------------------------------------------- /prototype/data/datasets/imagnetp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /prototype/data/image_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/image_reader.py -------------------------------------------------------------------------------- /prototype/data/imagenet_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/imagenet_dataloader.py -------------------------------------------------------------------------------- /prototype/data/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/metrics/__init__.py -------------------------------------------------------------------------------- /prototype/data/metrics/base_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/metrics/base_evaluator.py -------------------------------------------------------------------------------- /prototype/data/metrics/custom_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/metrics/custom_evaluator.py -------------------------------------------------------------------------------- /prototype/data/metrics/imagenet_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/metrics/imagenet_evaluator.py -------------------------------------------------------------------------------- /prototype/data/metrics/imagenetc_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/metrics/imagenetc_evaluator.py -------------------------------------------------------------------------------- /prototype/data/metrics/multiclass_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/metrics/multiclass_evaluator.py -------------------------------------------------------------------------------- /prototype/data/nvidia_dali_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/nvidia_dali_dataloader.py -------------------------------------------------------------------------------- /prototype/data/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/pipelines/__init__.py -------------------------------------------------------------------------------- /prototype/data/pipelines/imagenet_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/pipelines/imagenet_pipeline.py -------------------------------------------------------------------------------- /prototype/data/pipelines/imagenet_pipeline_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/pipelines/imagenet_pipeline_v2.py -------------------------------------------------------------------------------- /prototype/data/register_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/register_factory.py -------------------------------------------------------------------------------- /prototype/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/sampler.py -------------------------------------------------------------------------------- /prototype/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/transforms.py -------------------------------------------------------------------------------- /prototype/data/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /prototype/data/utils/calibration_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/utils/calibration_tools.py -------------------------------------------------------------------------------- /prototype/data/utils/video_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/data/utils/video_loader.py -------------------------------------------------------------------------------- /prototype/loss_functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/loss_functions/__init__.py -------------------------------------------------------------------------------- /prototype/loss_functions/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/loss_functions/loss.py -------------------------------------------------------------------------------- /prototype/loss_functions/nt_xent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/loss_functions/nt_xent.py -------------------------------------------------------------------------------- /prototype/lr_scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/lr_scheduler/__init__.py -------------------------------------------------------------------------------- /prototype/lr_scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/lr_scheduler/scheduler.py -------------------------------------------------------------------------------- /prototype/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/__init__.py -------------------------------------------------------------------------------- /prototype/model/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/alexnet.py -------------------------------------------------------------------------------- /prototype/model/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/densenet.py -------------------------------------------------------------------------------- /prototype/model/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/efficientnet.py -------------------------------------------------------------------------------- /prototype/model/layer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/layer/__init__.py -------------------------------------------------------------------------------- /prototype/model/layer/condconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/layer/condconv.py -------------------------------------------------------------------------------- /prototype/model/layer/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/layer/drop_path.py -------------------------------------------------------------------------------- /prototype/model/layer/sparse_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/layer/sparse_layer.py -------------------------------------------------------------------------------- /prototype/model/layer/splat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/layer/splat.py -------------------------------------------------------------------------------- /prototype/model/layer/weightnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/layer/weightnet.py -------------------------------------------------------------------------------- /prototype/model/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/mobilenet_v2.py -------------------------------------------------------------------------------- /prototype/model/mobilenet_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/mobilenet_v3.py -------------------------------------------------------------------------------- /prototype/model/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/regnet.py -------------------------------------------------------------------------------- /prototype/model/repvgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/repvgg.py -------------------------------------------------------------------------------- /prototype/model/resnet_official.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/resnet_official.py -------------------------------------------------------------------------------- /prototype/model/shufflenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/shufflenet_v2.py -------------------------------------------------------------------------------- /prototype/model/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vision_transformer.py -------------------------------------------------------------------------------- /prototype/model/vit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/__init__.py -------------------------------------------------------------------------------- /prototype/model/vit/default_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/default_cfg.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/__init__.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/attention/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/attention/__init__.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/attention/levit_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/attention/levit_attention.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/attention/swin_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/attention/swin_attention.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/attention/token2token_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/attention/token2token_attention.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/attention/vanilla_msa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/attention/vanilla_msa.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/drop_path.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/helpers.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/mlp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/mlp/__init__.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/mlp/vanilla_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/mlp/vanilla_mlp.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/patch_embedding/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/patch_embedding/__init__.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/patch_embedding/levit_hybrid_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/patch_embedding/levit_hybrid_backbone.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/patch_embedding/token2token_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/patch_embedding/token2token_embedding.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/patch_embedding/vanilla_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/patch_embedding/vanilla_embedding.py -------------------------------------------------------------------------------- /prototype/model/vit/layers/position_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/layers/position_embedding.py -------------------------------------------------------------------------------- /prototype/model/vit/mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/mlp_mixer.py -------------------------------------------------------------------------------- /prototype/model/vit/swin_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/swin_loader.py -------------------------------------------------------------------------------- /prototype/model/vit/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/swin_transformer.py -------------------------------------------------------------------------------- /prototype/model/vit/token2token_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/token2token_vit.py -------------------------------------------------------------------------------- /prototype/model/vit/vit_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/model/vit/vit_base.py -------------------------------------------------------------------------------- /prototype/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/optimizer/__init__.py -------------------------------------------------------------------------------- /prototype/solver/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/README.md -------------------------------------------------------------------------------- /prototype/solver/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /prototype/solver/adv_cls_solver_train_pgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/adv_cls_solver_train_pgd.py -------------------------------------------------------------------------------- /prototype/solver/base_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/base_solver.py -------------------------------------------------------------------------------- /prototype/solver/benchmark_eval_adv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/benchmark_eval_adv.py -------------------------------------------------------------------------------- /prototype/solver/cls_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/cls_solver.py -------------------------------------------------------------------------------- /prototype/solver/cls_solver_robust.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/cls_solver_robust.py -------------------------------------------------------------------------------- /prototype/solver/imgnet_a_o_eval_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/imgnet_a_o_eval_solver.py -------------------------------------------------------------------------------- /prototype/solver/imgnet_c_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/imgnet_c_solver.py -------------------------------------------------------------------------------- /prototype/solver/imgnet_p_eval_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/imgnet_p_eval_solver.py -------------------------------------------------------------------------------- /prototype/solver/mimic_eval_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/mimic_eval_solver.py -------------------------------------------------------------------------------- /prototype/solver/multi_eval_decoder_resize_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/multi_eval_decoder_resize_solver.py -------------------------------------------------------------------------------- /prototype/solver/reparam_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/reparam_solver.py -------------------------------------------------------------------------------- /prototype/solver/vit_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/solver/vit_solver.py -------------------------------------------------------------------------------- /prototype/tools/Draw_resize_diff.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/tools/Draw_resize_diff.ipynb -------------------------------------------------------------------------------- /prototype/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /prototype/tools/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/tools/inference.py -------------------------------------------------------------------------------- /prototype/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /prototype/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/utils/dist.py -------------------------------------------------------------------------------- /prototype/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/utils/ema.py -------------------------------------------------------------------------------- /prototype/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/utils/misc.py -------------------------------------------------------------------------------- /prototype/utils/model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/utils/model_config.py -------------------------------------------------------------------------------- /prototype/utils/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/utils/register.py -------------------------------------------------------------------------------- /prototype/utils/trunc_normal_initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/prototype/utils/trunc_normal_initializer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/requirements.txt -------------------------------------------------------------------------------- /spring/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /spring/linklink/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/spring/linklink/__init__.py -------------------------------------------------------------------------------- /spring/linklink/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Prototype/HEAD/spring/linklink/nn.py --------------------------------------------------------------------------------