├── .flake8 ├── .gitignore ├── .pre-commit-config.yaml ├── README.md ├── docs └── 深度学习tricks-我的一点个人见解.md ├── examples └── kaggle_leaves_classification.ipynb ├── requirements.txt ├── setup.py ├── tests └── unit_test │ ├── test_layers.py │ ├── test_losses.py │ └── test_models.py └── torch_utils ├── __init__.py ├── advanced ├── NextVLAD.py ├── __init__.py ├── arcface.py └── dolg.py ├── criterion ├── __init__.py ├── bitempered_loss.py ├── cross_entropy.py ├── dice.py ├── focal.py ├── lovasz.py ├── metric_loss.py └── rmi.py ├── dataset ├── __init__.py ├── common_aug.py ├── customized_aug.py ├── data_all_in_gpu.py ├── dataloader.py ├── del_duplicate_image.py ├── mixup.py ├── randaugment.py ├── random.py └── visualize.py ├── lr_scheduler ├── __init__.py ├── concat.py ├── cosine_annealing_with_warmup.py ├── customized.py └── onecycle.py ├── models ├── __init__.py ├── backbone │ ├── __init__.py │ └── timm_models.py ├── cls_models │ ├── __init__.py │ ├── hybrid_cls_model.py │ └── simple_cls_model.py ├── layers │ ├── __init__.py │ ├── anti_alias.py │ └── layers.py ├── seg_models │ ├── __init__.py │ └── unet.py └── utils │ ├── __init__.py │ └── freeze_norm.py ├── optimizer ├── __init__.py ├── gc.py ├── group_optim.py ├── lookahead.py ├── over9000.py ├── radam.py ├── ranger.py ├── ranger21 │ ├── __init__.py │ ├── chebyshev_lr_functions.py │ ├── ranger21.py │ └── rangerabel.py └── timm_optim │ ├── __init__.py │ ├── lamb.py │ └── lars.py └── tools.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/README.md -------------------------------------------------------------------------------- /docs/深度学习tricks-我的一点个人见解.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/kaggle_leaves_classification.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/examples/kaggle_leaves_classification.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/setup.py -------------------------------------------------------------------------------- /tests/unit_test/test_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/tests/unit_test/test_layers.py -------------------------------------------------------------------------------- /tests/unit_test/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/tests/unit_test/test_losses.py -------------------------------------------------------------------------------- /tests/unit_test/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/tests/unit_test/test_models.py -------------------------------------------------------------------------------- /torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/__init__.py -------------------------------------------------------------------------------- /torch_utils/advanced/NextVLAD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/advanced/NextVLAD.py -------------------------------------------------------------------------------- /torch_utils/advanced/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/advanced/__init__.py -------------------------------------------------------------------------------- /torch_utils/advanced/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/advanced/arcface.py -------------------------------------------------------------------------------- /torch_utils/advanced/dolg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/advanced/dolg.py -------------------------------------------------------------------------------- /torch_utils/criterion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/criterion/__init__.py -------------------------------------------------------------------------------- /torch_utils/criterion/bitempered_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/criterion/bitempered_loss.py -------------------------------------------------------------------------------- /torch_utils/criterion/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/criterion/cross_entropy.py -------------------------------------------------------------------------------- /torch_utils/criterion/dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/criterion/dice.py -------------------------------------------------------------------------------- /torch_utils/criterion/focal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/criterion/focal.py -------------------------------------------------------------------------------- /torch_utils/criterion/lovasz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/criterion/lovasz.py -------------------------------------------------------------------------------- /torch_utils/criterion/metric_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/criterion/metric_loss.py -------------------------------------------------------------------------------- /torch_utils/criterion/rmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/criterion/rmi.py -------------------------------------------------------------------------------- /torch_utils/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/__init__.py -------------------------------------------------------------------------------- /torch_utils/dataset/common_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/common_aug.py -------------------------------------------------------------------------------- /torch_utils/dataset/customized_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/customized_aug.py -------------------------------------------------------------------------------- /torch_utils/dataset/data_all_in_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/data_all_in_gpu.py -------------------------------------------------------------------------------- /torch_utils/dataset/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/dataloader.py -------------------------------------------------------------------------------- /torch_utils/dataset/del_duplicate_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/del_duplicate_image.py -------------------------------------------------------------------------------- /torch_utils/dataset/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/mixup.py -------------------------------------------------------------------------------- /torch_utils/dataset/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/randaugment.py -------------------------------------------------------------------------------- /torch_utils/dataset/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/random.py -------------------------------------------------------------------------------- /torch_utils/dataset/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/dataset/visualize.py -------------------------------------------------------------------------------- /torch_utils/lr_scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/lr_scheduler/__init__.py -------------------------------------------------------------------------------- /torch_utils/lr_scheduler/concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/lr_scheduler/concat.py -------------------------------------------------------------------------------- /torch_utils/lr_scheduler/cosine_annealing_with_warmup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/lr_scheduler/cosine_annealing_with_warmup.py -------------------------------------------------------------------------------- /torch_utils/lr_scheduler/customized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/lr_scheduler/customized.py -------------------------------------------------------------------------------- /torch_utils/lr_scheduler/onecycle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/lr_scheduler/onecycle.py -------------------------------------------------------------------------------- /torch_utils/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/__init__.py -------------------------------------------------------------------------------- /torch_utils/models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/backbone/__init__.py -------------------------------------------------------------------------------- /torch_utils/models/backbone/timm_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/backbone/timm_models.py -------------------------------------------------------------------------------- /torch_utils/models/cls_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/cls_models/__init__.py -------------------------------------------------------------------------------- /torch_utils/models/cls_models/hybrid_cls_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/cls_models/hybrid_cls_model.py -------------------------------------------------------------------------------- /torch_utils/models/cls_models/simple_cls_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/cls_models/simple_cls_model.py -------------------------------------------------------------------------------- /torch_utils/models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/layers/__init__.py -------------------------------------------------------------------------------- /torch_utils/models/layers/anti_alias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/layers/anti_alias.py -------------------------------------------------------------------------------- /torch_utils/models/layers/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/layers/layers.py -------------------------------------------------------------------------------- /torch_utils/models/seg_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/seg_models/__init__.py -------------------------------------------------------------------------------- /torch_utils/models/seg_models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/seg_models/unet.py -------------------------------------------------------------------------------- /torch_utils/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/utils/__init__.py -------------------------------------------------------------------------------- /torch_utils/models/utils/freeze_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/models/utils/freeze_norm.py -------------------------------------------------------------------------------- /torch_utils/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/__init__.py -------------------------------------------------------------------------------- /torch_utils/optimizer/gc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/gc.py -------------------------------------------------------------------------------- /torch_utils/optimizer/group_optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/group_optim.py -------------------------------------------------------------------------------- /torch_utils/optimizer/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/lookahead.py -------------------------------------------------------------------------------- /torch_utils/optimizer/over9000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/over9000.py -------------------------------------------------------------------------------- /torch_utils/optimizer/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/radam.py -------------------------------------------------------------------------------- /torch_utils/optimizer/ranger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/ranger.py -------------------------------------------------------------------------------- /torch_utils/optimizer/ranger21/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/ranger21/__init__.py -------------------------------------------------------------------------------- /torch_utils/optimizer/ranger21/chebyshev_lr_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/ranger21/chebyshev_lr_functions.py -------------------------------------------------------------------------------- /torch_utils/optimizer/ranger21/ranger21.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/ranger21/ranger21.py -------------------------------------------------------------------------------- /torch_utils/optimizer/ranger21/rangerabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/ranger21/rangerabel.py -------------------------------------------------------------------------------- /torch_utils/optimizer/timm_optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/timm_optim/__init__.py -------------------------------------------------------------------------------- /torch_utils/optimizer/timm_optim/lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/timm_optim/lamb.py -------------------------------------------------------------------------------- /torch_utils/optimizer/timm_optim/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/optimizer/timm_optim/lars.py -------------------------------------------------------------------------------- /torch_utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seefun/TorchUtils/HEAD/torch_utils/tools.py --------------------------------------------------------------------------------