├── .gitignore ├── LICENSE ├── MODEL_ZOO.md ├── README.md ├── configs ├── archive │ ├── cifar │ │ ├── anynet │ │ │ ├── R-110_nds_1gpu.yaml │ │ │ ├── R-56_nds_1gpu.yaml │ │ │ └── V-56_nds_1gpu.yaml │ │ └── resnet │ │ │ ├── R-110_nds_1gpu.yaml │ │ │ └── R-56_nds_1gpu.yaml │ └── imagenet │ │ ├── anynet │ │ ├── R-101-1x64d_step_1gpu.yaml │ │ ├── R-101-1x64d_step_2gpu.yaml │ │ ├── R-101-1x64d_step_8gpu.yaml │ │ ├── R-50-1x64d_step_1gpu.yaml │ │ ├── R-50-1x64d_step_2gpu.yaml │ │ ├── R-50-1x64d_step_8gpu.yaml │ │ ├── X-101-32x4d_step_1gpu.yaml │ │ ├── X-101-32x4d_step_2gpu.yaml │ │ ├── X-101-32x4d_step_8gpu.yaml │ │ ├── X-50-32x4d_step_1gpu.yaml │ │ ├── X-50-32x4d_step_2gpu.yaml │ │ └── X-50-32x4d_step_8gpu.yaml │ │ └── resnet │ │ ├── R-101-1x64d_step_1gpu.yaml │ │ ├── R-101-1x64d_step_2gpu.yaml │ │ ├── R-101-1x64d_step_8gpu.yaml │ │ ├── R-50-1x64d_step_1gpu.yaml │ │ ├── R-50-1x64d_step_2gpu.yaml │ │ ├── R-50-1x64d_step_8gpu.yaml │ │ ├── X-101-32x4d_step_1gpu.yaml │ │ ├── X-101-32x4d_step_2gpu.yaml │ │ ├── X-101-32x4d_step_8gpu.yaml │ │ ├── X-50-32x4d_step_1gpu.yaml │ │ ├── X-50-32x4d_step_2gpu.yaml │ │ └── X-50-32x4d_step_8gpu.yaml ├── dds_baselines │ ├── effnet │ │ ├── EN-B0_dds_8gpu.yaml │ │ ├── EN-B1_dds_8gpu.yaml │ │ ├── EN-B2_dds_8gpu.yaml │ │ ├── EN-B3_dds_8gpu.yaml │ │ ├── EN-B4_dds_8gpu.yaml │ │ └── EN-B5_dds_8gpu.yaml │ ├── regnetx │ │ ├── RegNetX-1.6GF_dds_8gpu.yaml │ │ ├── RegNetX-12GF_dds_8gpu.yaml │ │ ├── RegNetX-16GF_dds_8gpu.yaml │ │ ├── RegNetX-200MF_dds_8gpu.yaml │ │ ├── RegNetX-3.2GF_dds_8gpu.yaml │ │ ├── RegNetX-32GF_dds_8gpu.yaml │ │ ├── RegNetX-4.0GF_dds_8gpu.yaml │ │ ├── RegNetX-400MF_dds_8gpu.yaml │ │ ├── RegNetX-6.4GF_dds_8gpu.yaml │ │ ├── RegNetX-600MF_dds_8gpu.yaml │ │ ├── RegNetX-8.0GF_dds_8gpu.yaml │ │ └── RegNetX-800MF_dds_8gpu.yaml │ ├── regnety │ │ ├── RegNetY-1.6GF_dds_8gpu.yaml │ │ ├── RegNetY-12GF_dds_8gpu.yaml │ │ ├── RegNetY-16GF_dds_8gpu.yaml │ │ ├── RegNetY-200MF_dds_8gpu.yaml │ │ ├── RegNetY-3.2GF_dds_8gpu.yaml │ │ ├── RegNetY-32GF_dds_8gpu.yaml │ │ ├── RegNetY-4.0GF_dds_8gpu.yaml │ │ ├── RegNetY-400MF_dds_8gpu.yaml │ │ ├── RegNetY-6.4GF_dds_8gpu.yaml │ │ ├── RegNetY-600MF_dds_8gpu.yaml │ │ ├── RegNetY-8.0GF_dds_8gpu.yaml │ │ └── RegNetY-800MF_dds_8gpu.yaml │ ├── resnet │ │ ├── R-101-1x64d_dds_8gpu.yaml │ │ ├── R-152-1x64d_dds_8gpu.yaml │ │ └── R-50-1x64d_dds_8gpu.yaml │ └── resnext │ │ ├── X-101-32x4d_dds_8gpu.yaml │ │ ├── X-152-32x4d_dds_8gpu.yaml │ │ └── X-50-32x4d_dds_8gpu.yaml ├── examples │ └── ffcv.yaml ├── fsdp │ └── example.yaml └── sweeps │ └── cifar │ ├── cifar_best.yaml │ ├── cifar_optim.yaml │ └── cifar_regnet.yaml ├── dev ├── .flake8 ├── .isort.cfg ├── linter.sh ├── model_complexity.json ├── model_error.json ├── model_timing.json ├── model_zoo_tables.py └── test_models.py ├── docs ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── DATA.md ├── GETTING_STARTED.md ├── SWEEP_INFO.md └── regnetx_nets.png ├── pycls ├── __init__.py ├── core │ ├── __init__.py │ ├── benchmark.py │ ├── builders.py │ ├── checkpoint.py │ ├── config.py │ ├── distributed.py │ ├── io.py │ ├── logging.py │ ├── meters.py │ ├── net.py │ ├── optimizer.py │ ├── plotting.py │ ├── timer.py │ └── trainer.py ├── datasets │ ├── __init__.py │ ├── augment.py │ ├── cifar10.py │ ├── imagenet.py │ ├── loader.py │ └── transforms.py ├── models │ ├── __init__.py │ ├── anynet.py │ ├── blocks.py │ ├── effnet.py │ ├── model_zoo.py │ ├── regnet.py │ ├── resnet.py │ ├── scaler.py │ └── vit.py └── sweep │ ├── __init__.py │ ├── analysis.py │ ├── config.py │ ├── htmlbook.py │ ├── plotting.py │ ├── random.py │ └── samplers.py ├── requirements.txt ├── setup.py └── tools ├── run_net.py ├── sweep_analyze.py ├── sweep_collect.py ├── sweep_launch.py ├── sweep_launch_job.py └── sweep_setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_ZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/MODEL_ZOO.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/README.md -------------------------------------------------------------------------------- /configs/archive/cifar/anynet/R-110_nds_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/cifar/anynet/R-110_nds_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/cifar/anynet/R-56_nds_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/cifar/anynet/R-56_nds_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/cifar/anynet/V-56_nds_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/cifar/anynet/V-56_nds_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/cifar/resnet/R-110_nds_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/cifar/resnet/R-110_nds_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/cifar/resnet/R-56_nds_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/cifar/resnet/R-56_nds_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/R-101-1x64d_step_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/R-101-1x64d_step_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/R-101-1x64d_step_2gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/R-101-1x64d_step_2gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/R-101-1x64d_step_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/R-101-1x64d_step_8gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/R-50-1x64d_step_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/R-50-1x64d_step_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/R-50-1x64d_step_2gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/R-50-1x64d_step_2gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/R-50-1x64d_step_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/R-50-1x64d_step_8gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/X-101-32x4d_step_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/X-101-32x4d_step_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/X-101-32x4d_step_2gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/X-101-32x4d_step_2gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/X-101-32x4d_step_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/X-101-32x4d_step_8gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/X-50-32x4d_step_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/X-50-32x4d_step_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/X-50-32x4d_step_2gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/X-50-32x4d_step_2gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/anynet/X-50-32x4d_step_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/anynet/X-50-32x4d_step_8gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/R-101-1x64d_step_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/R-101-1x64d_step_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/R-101-1x64d_step_2gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/R-101-1x64d_step_2gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/R-101-1x64d_step_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/R-101-1x64d_step_8gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/R-50-1x64d_step_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/R-50-1x64d_step_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/R-50-1x64d_step_2gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/R-50-1x64d_step_2gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/R-50-1x64d_step_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/R-50-1x64d_step_8gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/X-101-32x4d_step_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/X-101-32x4d_step_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/X-101-32x4d_step_2gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/X-101-32x4d_step_2gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/X-101-32x4d_step_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/X-101-32x4d_step_8gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/X-50-32x4d_step_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/X-50-32x4d_step_1gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/X-50-32x4d_step_2gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/X-50-32x4d_step_2gpu.yaml -------------------------------------------------------------------------------- /configs/archive/imagenet/resnet/X-50-32x4d_step_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/archive/imagenet/resnet/X-50-32x4d_step_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/effnet/EN-B0_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/effnet/EN-B0_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/effnet/EN-B1_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/effnet/EN-B1_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/effnet/EN-B2_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/effnet/EN-B2_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/effnet/EN-B3_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/effnet/EN-B3_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/effnet/EN-B4_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/effnet/EN-B4_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/effnet/EN-B5_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/effnet/EN-B5_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-1.6GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-1.6GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-12GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-12GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-16GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-16GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-200MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-200MF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-3.2GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-3.2GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-32GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-32GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-4.0GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-4.0GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-400MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-400MF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-6.4GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-6.4GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-600MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-600MF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-8.0GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-8.0GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnetx/RegNetX-800MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnetx/RegNetX-800MF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-1.6GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-1.6GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-12GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-12GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-16GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-16GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-200MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-200MF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-3.2GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-3.2GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-32GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-32GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-4.0GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-4.0GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-400MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-400MF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-6.4GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-6.4GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-600MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-600MF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-8.0GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-8.0GF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/regnety/RegNetY-800MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/regnety/RegNetY-800MF_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/resnet/R-101-1x64d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/resnet/R-101-1x64d_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/resnet/R-152-1x64d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/resnet/R-152-1x64d_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/resnet/R-50-1x64d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/resnet/R-50-1x64d_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/resnext/X-101-32x4d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/resnext/X-101-32x4d_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/resnext/X-152-32x4d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/resnext/X-152-32x4d_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/dds_baselines/resnext/X-50-32x4d_dds_8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/dds_baselines/resnext/X-50-32x4d_dds_8gpu.yaml -------------------------------------------------------------------------------- /configs/examples/ffcv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/examples/ffcv.yaml -------------------------------------------------------------------------------- /configs/fsdp/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/fsdp/example.yaml -------------------------------------------------------------------------------- /configs/sweeps/cifar/cifar_best.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/sweeps/cifar/cifar_best.yaml -------------------------------------------------------------------------------- /configs/sweeps/cifar/cifar_optim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/sweeps/cifar/cifar_optim.yaml -------------------------------------------------------------------------------- /configs/sweeps/cifar/cifar_regnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/configs/sweeps/cifar/cifar_regnet.yaml -------------------------------------------------------------------------------- /dev/.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/dev/.flake8 -------------------------------------------------------------------------------- /dev/.isort.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/dev/.isort.cfg -------------------------------------------------------------------------------- /dev/linter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/dev/linter.sh -------------------------------------------------------------------------------- /dev/model_complexity.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/dev/model_complexity.json -------------------------------------------------------------------------------- /dev/model_error.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/dev/model_error.json -------------------------------------------------------------------------------- /dev/model_timing.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/dev/model_timing.json -------------------------------------------------------------------------------- /dev/model_zoo_tables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/dev/model_zoo_tables.py -------------------------------------------------------------------------------- /dev/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/dev/test_models.py -------------------------------------------------------------------------------- /docs/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/docs/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /docs/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/docs/CONTRIBUTING.md -------------------------------------------------------------------------------- /docs/DATA.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/docs/DATA.md -------------------------------------------------------------------------------- /docs/GETTING_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/docs/GETTING_STARTED.md -------------------------------------------------------------------------------- /docs/SWEEP_INFO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/docs/SWEEP_INFO.md -------------------------------------------------------------------------------- /docs/regnetx_nets.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/docs/regnetx_nets.png -------------------------------------------------------------------------------- /pycls/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pycls/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pycls/core/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/benchmark.py -------------------------------------------------------------------------------- /pycls/core/builders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/builders.py -------------------------------------------------------------------------------- /pycls/core/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/checkpoint.py -------------------------------------------------------------------------------- /pycls/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/config.py -------------------------------------------------------------------------------- /pycls/core/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/distributed.py -------------------------------------------------------------------------------- /pycls/core/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/io.py -------------------------------------------------------------------------------- /pycls/core/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/logging.py -------------------------------------------------------------------------------- /pycls/core/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/meters.py -------------------------------------------------------------------------------- /pycls/core/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/net.py -------------------------------------------------------------------------------- /pycls/core/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/optimizer.py -------------------------------------------------------------------------------- /pycls/core/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/plotting.py -------------------------------------------------------------------------------- /pycls/core/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/timer.py -------------------------------------------------------------------------------- /pycls/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/core/trainer.py -------------------------------------------------------------------------------- /pycls/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pycls/datasets/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/datasets/augment.py -------------------------------------------------------------------------------- /pycls/datasets/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/datasets/cifar10.py -------------------------------------------------------------------------------- /pycls/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/datasets/imagenet.py -------------------------------------------------------------------------------- /pycls/datasets/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/datasets/loader.py -------------------------------------------------------------------------------- /pycls/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/datasets/transforms.py -------------------------------------------------------------------------------- /pycls/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/models/__init__.py -------------------------------------------------------------------------------- /pycls/models/anynet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/models/anynet.py -------------------------------------------------------------------------------- /pycls/models/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/models/blocks.py -------------------------------------------------------------------------------- /pycls/models/effnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/models/effnet.py -------------------------------------------------------------------------------- /pycls/models/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/models/model_zoo.py -------------------------------------------------------------------------------- /pycls/models/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/models/regnet.py -------------------------------------------------------------------------------- /pycls/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/models/resnet.py -------------------------------------------------------------------------------- /pycls/models/scaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/models/scaler.py -------------------------------------------------------------------------------- /pycls/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/models/vit.py -------------------------------------------------------------------------------- /pycls/sweep/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pycls/sweep/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/sweep/analysis.py -------------------------------------------------------------------------------- /pycls/sweep/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/sweep/config.py -------------------------------------------------------------------------------- /pycls/sweep/htmlbook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/sweep/htmlbook.py -------------------------------------------------------------------------------- /pycls/sweep/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/sweep/plotting.py -------------------------------------------------------------------------------- /pycls/sweep/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/sweep/random.py -------------------------------------------------------------------------------- /pycls/sweep/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/pycls/sweep/samplers.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/setup.py -------------------------------------------------------------------------------- /tools/run_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/tools/run_net.py -------------------------------------------------------------------------------- /tools/sweep_analyze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/tools/sweep_analyze.py -------------------------------------------------------------------------------- /tools/sweep_collect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/tools/sweep_collect.py -------------------------------------------------------------------------------- /tools/sweep_launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/tools/sweep_launch.py -------------------------------------------------------------------------------- /tools/sweep_launch_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/tools/sweep_launch_job.py -------------------------------------------------------------------------------- /tools/sweep_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/pycls/HEAD/tools/sweep_setup.py --------------------------------------------------------------------------------