├── .gitignore ├── CHANGELOG ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── data ├── distractors.txt └── whiten.txt ├── environment.yml ├── multigrain ├── __init__.py ├── augmentations │ ├── __init__.py │ ├── autoaugment.py │ └── transforms.py ├── backbones │ ├── __init__.py │ ├── backbone.py │ ├── nasnet_mobile.py │ └── pnasnet.py ├── datasets │ ├── __init__.py │ ├── holidays-rotate.yaml │ ├── id_dataset.py │ ├── imagenet.py │ ├── list_dataset.py │ ├── loader.py │ └── retrieval.py ├── lib │ ├── __init__.py │ ├── multigrain.py │ ├── samplers.py │ └── whiten.py ├── modules │ ├── __init__.py │ ├── criterion.py │ ├── functional.py │ ├── layers.py │ ├── margin.py │ └── multioptim.py └── utils │ ├── __init__.py │ ├── arguments.py │ ├── checkpoint.py │ ├── logging.py │ ├── metrics.py │ ├── misc.py │ ├── plots.py │ ├── tictoc.py │ └── torch_utils.py ├── requirements.txt └── scripts ├── .gitignore ├── evaluate.py ├── finetune_p.py ├── set_path.py ├── train.py └── whiten.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG: -------------------------------------------------------------------------------- 1 | --- 2 | -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/README.md -------------------------------------------------------------------------------- /data/distractors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/data/distractors.txt -------------------------------------------------------------------------------- /data/whiten.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/data/whiten.txt -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/environment.yml -------------------------------------------------------------------------------- /multigrain/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multigrain/augmentations/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/augmentations/__init__.py -------------------------------------------------------------------------------- /multigrain/augmentations/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/augmentations/autoaugment.py -------------------------------------------------------------------------------- /multigrain/augmentations/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/augmentations/transforms.py -------------------------------------------------------------------------------- /multigrain/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/backbones/__init__.py -------------------------------------------------------------------------------- /multigrain/backbones/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/backbones/backbone.py -------------------------------------------------------------------------------- /multigrain/backbones/nasnet_mobile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/backbones/nasnet_mobile.py -------------------------------------------------------------------------------- /multigrain/backbones/pnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/backbones/pnasnet.py -------------------------------------------------------------------------------- /multigrain/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/datasets/__init__.py -------------------------------------------------------------------------------- /multigrain/datasets/holidays-rotate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/datasets/holidays-rotate.yaml -------------------------------------------------------------------------------- /multigrain/datasets/id_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/datasets/id_dataset.py -------------------------------------------------------------------------------- /multigrain/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/datasets/imagenet.py -------------------------------------------------------------------------------- /multigrain/datasets/list_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/datasets/list_dataset.py -------------------------------------------------------------------------------- /multigrain/datasets/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/datasets/loader.py -------------------------------------------------------------------------------- /multigrain/datasets/retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/datasets/retrieval.py -------------------------------------------------------------------------------- /multigrain/lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/lib/__init__.py -------------------------------------------------------------------------------- /multigrain/lib/multigrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/lib/multigrain.py -------------------------------------------------------------------------------- /multigrain/lib/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/lib/samplers.py -------------------------------------------------------------------------------- /multigrain/lib/whiten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/lib/whiten.py -------------------------------------------------------------------------------- /multigrain/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/modules/__init__.py -------------------------------------------------------------------------------- /multigrain/modules/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/modules/criterion.py -------------------------------------------------------------------------------- /multigrain/modules/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/modules/functional.py -------------------------------------------------------------------------------- /multigrain/modules/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/modules/layers.py -------------------------------------------------------------------------------- /multigrain/modules/margin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/modules/margin.py -------------------------------------------------------------------------------- /multigrain/modules/multioptim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/modules/multioptim.py -------------------------------------------------------------------------------- /multigrain/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/utils/__init__.py -------------------------------------------------------------------------------- /multigrain/utils/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/utils/arguments.py -------------------------------------------------------------------------------- /multigrain/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/utils/checkpoint.py -------------------------------------------------------------------------------- /multigrain/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/utils/logging.py -------------------------------------------------------------------------------- /multigrain/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/utils/metrics.py -------------------------------------------------------------------------------- /multigrain/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/utils/misc.py -------------------------------------------------------------------------------- /multigrain/utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/utils/plots.py -------------------------------------------------------------------------------- /multigrain/utils/tictoc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/utils/tictoc.py -------------------------------------------------------------------------------- /multigrain/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/multigrain/utils/torch_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/.gitignore: -------------------------------------------------------------------------------- 1 | config_user.yaml 2 | -------------------------------------------------------------------------------- /scripts/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/scripts/evaluate.py -------------------------------------------------------------------------------- /scripts/finetune_p.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/scripts/finetune_p.py -------------------------------------------------------------------------------- /scripts/set_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/scripts/set_path.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/scripts/train.py -------------------------------------------------------------------------------- /scripts/whiten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/multigrain/HEAD/scripts/whiten.py --------------------------------------------------------------------------------