├── .gitignore ├── .pyre_configuration ├── .watchmanconfig ├── CHANGELOG.md ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── dev-requirements.txt ├── pyproject.toml ├── requirements.txt ├── setup.py └── torchrecipes ├── __init__.py ├── _internal_patches.py ├── audio ├── __init__.py └── source_separation │ ├── conf │ ├── __init__.py │ └── default_config.yaml │ ├── datamodule │ ├── __init__.py │ ├── librimix.py │ └── utils.py │ ├── loss │ ├── __init__.py │ ├── si_sdr.py │ └── utils.py │ ├── main.py │ ├── metrics │ ├── __init__.py │ └── sdr.py │ ├── module │ └── conv_tasnet.py │ └── tests │ └── test_main.py ├── core ├── __init__.py ├── base_train_app.py ├── conf │ ├── __init__.py │ ├── lr_scheduler │ │ ├── exponential.yaml │ │ ├── multi_step.yaml │ │ └── step.yaml │ ├── module │ │ ├── loss │ │ │ └── cross_entropy.yaml │ │ ├── lr_scheduler │ │ │ └── step_lr.yaml │ │ └── optim │ │ │ ├── adam.yaml │ │ │ ├── adamw.yaml │ │ │ └── sgd.yaml │ ├── optim │ │ ├── adamw.yaml │ │ └── sgd.yaml │ └── trainer │ │ ├── cpu.yaml │ │ ├── multi_gpu.yaml │ │ └── single_gpu.yaml ├── logger.py ├── test_utils │ ├── conf_utils.py │ └── test_base.py └── tests │ └── test_base_train_app.py ├── launcher ├── __init__.py ├── run.py └── tests │ └── test_env.py ├── paved_path ├── .gitignore ├── .lintrunner.toml ├── README.md ├── airflow │ ├── README.md │ ├── charnn_dag.py │ └── setup.sh ├── charnn │ ├── char_dataset.py │ ├── char_transform.py │ ├── combined_module.py │ ├── data │ │ └── input.txt │ ├── export.py │ ├── main.py │ ├── model.py │ ├── serve │ │ ├── deploy.sh │ │ └── handler.py │ ├── tests │ │ └── test_cli.py │ ├── trainer.py │ ├── trainer_config.yaml │ └── utils.py ├── docker │ ├── Dockerfile │ ├── README.md │ └── build.sh ├── requirements.txt └── tools │ └── linter │ ├── pip_init.py │ └── ufmt_linter.py ├── rec ├── README.MD ├── __init__.py ├── accelerators │ ├── __init__.py │ └── torchrec.py ├── datamodules │ ├── __init__.py │ ├── commons.py │ ├── criteo_datamodule.py │ ├── random_rec_datamodule.py │ ├── samplers │ │ ├── tests │ │ │ └── test_undersampler.py │ │ └── undersampler.py │ └── tests │ │ ├── __init__.py │ │ ├── test_criteo_datamodule.py │ │ ├── test_random_rec_datamodule.py │ │ └── utils.py ├── dlrm_main.py ├── modules │ ├── __init__.py │ ├── lightning_dlrm.py │ ├── tests │ │ ├── __init__.py │ │ ├── test_lightning_dlrm.py │ │ └── test_unsharded_lightning_dlrm.py │ └── unsharded_lightning_dlrm.py └── tests │ ├── __init__.py │ └── test_dlrm_main.py ├── tests └── test_version.py ├── text └── doc_classification │ ├── .torchxconfig │ ├── README.md │ ├── __init__.py │ ├── conf │ ├── __init__.py │ ├── datamodule │ │ ├── dataset │ │ │ └── sst2_dataset.yaml │ │ └── doc_classification_datamodule.yaml │ ├── default_config.yaml │ ├── module │ │ ├── model │ │ │ ├── xlmrbase_classifier.yaml │ │ │ └── xlmrbase_classifier_tiny.yaml │ │ └── optim │ │ │ └── adamw.yaml │ ├── tiny_model_full_config.yaml │ ├── tiny_model_mixed_config.yaml │ ├── trainer │ │ ├── cpu.yaml │ │ ├── multi_gpu.yaml │ │ └── single_gpu.yaml │ ├── transform │ │ ├── doc_classification_transform.yaml │ │ └── doc_classification_transform_tiny.yaml │ └── xlmrbase_sst2_config.yaml │ ├── datamodule │ └── doc_classification.py │ ├── main.py │ ├── module │ └── doc_classification.py │ ├── tests │ ├── common │ │ ├── __init__.py │ │ └── assets.py │ ├── data │ │ ├── SST2 │ │ │ └── SST-2.zip │ │ ├── spm_example.model │ │ └── vocab_example.pt │ ├── test_doc_classification_config.py │ ├── test_doc_classification_datamodule.py │ ├── test_doc_classification_main.py │ ├── test_doc_classification_module.py │ └── test_doc_classification_transform.py │ └── transform │ └── doc_classification_text_transform.py ├── utils ├── __init__.py ├── checkpoint.py ├── config_utils.py ├── distributed_utils.py ├── mixup_utils.py ├── task_test_base.py ├── test.py ├── tests │ ├── test_config_utils.py │ └── test_trainer_plugins.py └── trainer_plugins.py ├── version.py └── vision ├── core ├── __init__.py ├── conf │ └── datamodule │ │ ├── mnist_datamodule.yaml │ │ └── torchvision_datamodule.yaml ├── datamodule │ ├── __init__.py │ ├── mnist_data_module.py │ ├── tests │ │ ├── test_mnist_data_module.py │ │ └── test_torchvision_data_module.py │ ├── torchvision_data_module.py │ └── transforms │ │ ├── __init__.py │ │ └── builder.py ├── ops │ ├── __init__.py │ └── fine_tuning_wrapper.py ├── optim │ └── lr_scheduler.py ├── tests │ ├── test_fine_tuning_wrapper.py │ └── test_lr_scheduler.py └── utils │ ├── lr_scheduler.py │ ├── model_weights.py │ ├── model_weights_exporter.py │ └── test_module.py ├── image_classification ├── README.md ├── __init__.py ├── callbacks │ ├── __init__.py │ └── mixup_transform.py ├── conf │ ├── __init__.py │ ├── datamodule │ │ ├── datasets │ │ │ ├── cifar10.yaml │ │ │ └── fake_data.yaml │ │ └── torchvision_datamodule.yaml │ ├── default_config.yaml │ ├── module │ │ ├── default_module.yaml │ │ ├── loss │ │ │ └── cross_entropy.yaml │ │ ├── lr_scheduler │ │ │ └── step_lr.yaml │ │ ├── metrics │ │ │ ├── accuracy.yaml │ │ │ ├── accuracy_top1_top5.yaml │ │ │ ├── average_precision.yaml │ │ │ └── multilabel_accuracy.yaml │ │ ├── model │ │ │ ├── resnet18.yaml │ │ │ ├── resnet50.yaml │ │ │ └── resnext101_32x4d.yaml │ │ └── optim │ │ │ └── sgd.yaml │ └── trainer │ │ ├── cpu.yaml │ │ └── gpu.yaml ├── losses │ ├── soft_target_cross_entropy_loss.py │ └── tests │ │ └── test_soft_target_cross_entropy_loss.py ├── main.py ├── metrics │ ├── multilabel_accuracy.py │ └── tests │ │ └── test_multilabel_accuracy.py ├── module │ └── image_classification.py └── tests │ ├── test_image_classification_module.py │ └── test_main.py └── image_generation ├── __init__.py ├── callbacks ├── __init__.py └── image_generation.py ├── conf ├── __init__.py ├── datamodule │ ├── datasets │ │ └── fake_data.yaml │ ├── mnist.yaml │ ├── torchvision_datamodule.yaml │ └── transforms │ │ └── resize.yaml ├── gan_train_app.yaml ├── infogan_train_app.yaml └── module │ ├── criterion │ └── bce_loss.yaml │ ├── discriminator │ ├── dcgan.yaml │ ├── gan.yaml │ └── infogan.yaml │ ├── generator │ ├── dcgan.yaml │ ├── gan.yaml │ └── infogan.yaml │ ├── optim │ ├── default.yaml │ └── infogan_default.yaml │ └── tests │ └── test_gan_module_conf.py ├── models ├── dcgan.py ├── gan.py └── infogan.py ├── module ├── gan.py └── infogan.py ├── tests ├── test_image_generation.py └── test_train_app.py └── train_app.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/.gitignore -------------------------------------------------------------------------------- /.pyre_configuration: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/.pyre_configuration -------------------------------------------------------------------------------- /.watchmanconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/.watchmanconfig -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- 1 | # CHANGELOG 2 | 3 | Summary of per-version changes. 4 | -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/README.md -------------------------------------------------------------------------------- /dev-requirements.txt: -------------------------------------------------------------------------------- 1 | black>=21.5b1 2 | importlib-metadata 3 | -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/setup.py -------------------------------------------------------------------------------- /torchrecipes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/__init__.py -------------------------------------------------------------------------------- /torchrecipes/_internal_patches.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/_internal_patches.py -------------------------------------------------------------------------------- /torchrecipes/audio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/__init__.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/conf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/conf/__init__.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/conf/default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/conf/default_config.yaml -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/datamodule/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/datamodule/__init__.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/datamodule/librimix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/datamodule/librimix.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/datamodule/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/datamodule/utils.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/loss/__init__.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/loss/si_sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/loss/si_sdr.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/loss/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/loss/utils.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/main.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/metrics/__init__.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/metrics/sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/metrics/sdr.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/module/conv_tasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/module/conv_tasnet.py -------------------------------------------------------------------------------- /torchrecipes/audio/source_separation/tests/test_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/audio/source_separation/tests/test_main.py -------------------------------------------------------------------------------- /torchrecipes/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/__init__.py -------------------------------------------------------------------------------- /torchrecipes/core/base_train_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/base_train_app.py -------------------------------------------------------------------------------- /torchrecipes/core/conf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/conf/__init__.py -------------------------------------------------------------------------------- /torchrecipes/core/conf/lr_scheduler/exponential.yaml: -------------------------------------------------------------------------------- 1 | _target_: "torch.optim.lr_scheduler.ExponentialLR" 2 | gamma: 0.1 3 | last_epoch: -1 4 | -------------------------------------------------------------------------------- /torchrecipes/core/conf/lr_scheduler/multi_step.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/conf/lr_scheduler/multi_step.yaml -------------------------------------------------------------------------------- /torchrecipes/core/conf/lr_scheduler/step.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/conf/lr_scheduler/step.yaml -------------------------------------------------------------------------------- /torchrecipes/core/conf/module/loss/cross_entropy.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.nn.CrossEntropyLoss 2 | -------------------------------------------------------------------------------- /torchrecipes/core/conf/module/lr_scheduler/step_lr.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.optim.lr_scheduler.StepLR 2 | step_size: 10 3 | gamma: 0.1 4 | -------------------------------------------------------------------------------- /torchrecipes/core/conf/module/optim/adam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/conf/module/optim/adam.yaml -------------------------------------------------------------------------------- /torchrecipes/core/conf/module/optim/adamw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/conf/module/optim/adamw.yaml -------------------------------------------------------------------------------- /torchrecipes/core/conf/module/optim/sgd.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.optim.SGD 2 | lr: 0.1 3 | weight_decay: 1e-4 4 | momentum: 0.9 5 | nesterov: False 6 | -------------------------------------------------------------------------------- /torchrecipes/core/conf/optim/adamw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/conf/optim/adamw.yaml -------------------------------------------------------------------------------- /torchrecipes/core/conf/optim/sgd.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.optim.SGD 2 | lr: 0.1 3 | weight_decay: 1e-4 4 | momentum: 0.9 5 | nesterov: True 6 | -------------------------------------------------------------------------------- /torchrecipes/core/conf/trainer/cpu.yaml: -------------------------------------------------------------------------------- 1 | max_epochs: 1 2 | -------------------------------------------------------------------------------- /torchrecipes/core/conf/trainer/multi_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/conf/trainer/multi_gpu.yaml -------------------------------------------------------------------------------- /torchrecipes/core/conf/trainer/single_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/conf/trainer/single_gpu.yaml -------------------------------------------------------------------------------- /torchrecipes/core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/logger.py -------------------------------------------------------------------------------- /torchrecipes/core/test_utils/conf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/test_utils/conf_utils.py -------------------------------------------------------------------------------- /torchrecipes/core/test_utils/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/test_utils/test_base.py -------------------------------------------------------------------------------- /torchrecipes/core/tests/test_base_train_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/core/tests/test_base_train_app.py -------------------------------------------------------------------------------- /torchrecipes/launcher/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/launcher/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/launcher/run.py -------------------------------------------------------------------------------- /torchrecipes/launcher/tests/test_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/launcher/tests/test_env.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/.gitignore: -------------------------------------------------------------------------------- 1 | # Hydra output directory 2 | .hydra 3 | -------------------------------------------------------------------------------- /torchrecipes/paved_path/.lintrunner.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/.lintrunner.toml -------------------------------------------------------------------------------- /torchrecipes/paved_path/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/README.md -------------------------------------------------------------------------------- /torchrecipes/paved_path/airflow/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/airflow/README.md -------------------------------------------------------------------------------- /torchrecipes/paved_path/airflow/charnn_dag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/airflow/charnn_dag.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/airflow/setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/airflow/setup.sh -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/char_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/char_dataset.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/char_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/char_transform.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/combined_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/combined_module.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/data/input.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/data/input.txt -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/export.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/main.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/model.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/serve/deploy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/serve/deploy.sh -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/serve/handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/serve/handler.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/tests/test_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/tests/test_cli.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/trainer.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/trainer_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/trainer_config.yaml -------------------------------------------------------------------------------- /torchrecipes/paved_path/charnn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/charnn/utils.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/docker/Dockerfile -------------------------------------------------------------------------------- /torchrecipes/paved_path/docker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/docker/README.md -------------------------------------------------------------------------------- /torchrecipes/paved_path/docker/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/docker/build.sh -------------------------------------------------------------------------------- /torchrecipes/paved_path/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/requirements.txt -------------------------------------------------------------------------------- /torchrecipes/paved_path/tools/linter/pip_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/tools/linter/pip_init.py -------------------------------------------------------------------------------- /torchrecipes/paved_path/tools/linter/ufmt_linter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/paved_path/tools/linter/ufmt_linter.py -------------------------------------------------------------------------------- /torchrecipes/rec/README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/README.MD -------------------------------------------------------------------------------- /torchrecipes/rec/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/__init__.py -------------------------------------------------------------------------------- /torchrecipes/rec/accelerators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/rec/accelerators/torchrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/accelerators/torchrec.py -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/commons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/datamodules/commons.py -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/criteo_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/datamodules/criteo_datamodule.py -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/random_rec_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/datamodules/random_rec_datamodule.py -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/samplers/tests/test_undersampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/datamodules/samplers/tests/test_undersampler.py -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/samplers/undersampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/datamodules/samplers/undersampler.py -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/tests/test_criteo_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/datamodules/tests/test_criteo_datamodule.py -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/tests/test_random_rec_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/datamodules/tests/test_random_rec_datamodule.py -------------------------------------------------------------------------------- /torchrecipes/rec/datamodules/tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/datamodules/tests/utils.py -------------------------------------------------------------------------------- /torchrecipes/rec/dlrm_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/dlrm_main.py -------------------------------------------------------------------------------- /torchrecipes/rec/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/rec/modules/lightning_dlrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/modules/lightning_dlrm.py -------------------------------------------------------------------------------- /torchrecipes/rec/modules/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/rec/modules/tests/test_lightning_dlrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/modules/tests/test_lightning_dlrm.py -------------------------------------------------------------------------------- /torchrecipes/rec/modules/tests/test_unsharded_lightning_dlrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/modules/tests/test_unsharded_lightning_dlrm.py -------------------------------------------------------------------------------- /torchrecipes/rec/modules/unsharded_lightning_dlrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/modules/unsharded_lightning_dlrm.py -------------------------------------------------------------------------------- /torchrecipes/rec/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/rec/tests/test_dlrm_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/rec/tests/test_dlrm_main.py -------------------------------------------------------------------------------- /torchrecipes/tests/test_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/tests/test_version.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/.torchxconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/.torchxconfig -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/README.md -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/__init__.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/datamodule/dataset/sst2_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/datamodule/dataset/sst2_dataset.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/datamodule/doc_classification_datamodule.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/datamodule/doc_classification_datamodule.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/default_config.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/module/model/xlmrbase_classifier.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/module/model/xlmrbase_classifier.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/module/model/xlmrbase_classifier_tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/module/model/xlmrbase_classifier_tiny.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/module/optim/adamw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/module/optim/adamw.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/tiny_model_full_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/tiny_model_full_config.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/tiny_model_mixed_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/tiny_model_mixed_config.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/trainer/cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/trainer/cpu.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/trainer/multi_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/trainer/multi_gpu.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/trainer/single_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/trainer/single_gpu.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/transform/doc_classification_transform.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/transform/doc_classification_transform.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/transform/doc_classification_transform_tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/transform/doc_classification_transform_tiny.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/conf/xlmrbase_sst2_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/conf/xlmrbase_sst2_config.yaml -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/datamodule/doc_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/datamodule/doc_classification.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/main.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/module/doc_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/module/doc_classification.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/common/assets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/tests/common/assets.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/data/SST2/SST-2.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/tests/data/SST2/SST-2.zip -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/data/spm_example.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/tests/data/spm_example.model -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/data/vocab_example.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/tests/data/vocab_example.pt -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/test_doc_classification_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/tests/test_doc_classification_config.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/test_doc_classification_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/tests/test_doc_classification_datamodule.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/test_doc_classification_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/tests/test_doc_classification_main.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/test_doc_classification_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/tests/test_doc_classification_module.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/tests/test_doc_classification_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/tests/test_doc_classification_transform.py -------------------------------------------------------------------------------- /torchrecipes/text/doc_classification/transform/doc_classification_text_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/text/doc_classification/transform/doc_classification_text_transform.py -------------------------------------------------------------------------------- /torchrecipes/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/__init__.py -------------------------------------------------------------------------------- /torchrecipes/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/checkpoint.py -------------------------------------------------------------------------------- /torchrecipes/utils/config_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/config_utils.py -------------------------------------------------------------------------------- /torchrecipes/utils/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/distributed_utils.py -------------------------------------------------------------------------------- /torchrecipes/utils/mixup_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/mixup_utils.py -------------------------------------------------------------------------------- /torchrecipes/utils/task_test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/task_test_base.py -------------------------------------------------------------------------------- /torchrecipes/utils/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/test.py -------------------------------------------------------------------------------- /torchrecipes/utils/tests/test_config_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/tests/test_config_utils.py -------------------------------------------------------------------------------- /torchrecipes/utils/tests/test_trainer_plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/tests/test_trainer_plugins.py -------------------------------------------------------------------------------- /torchrecipes/utils/trainer_plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/utils/trainer_plugins.py -------------------------------------------------------------------------------- /torchrecipes/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/version.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/vision/core/conf/datamodule/mnist_datamodule.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/conf/datamodule/mnist_datamodule.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/core/conf/datamodule/torchvision_datamodule.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/conf/datamodule/torchvision_datamodule.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/core/datamodule/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/datamodule/__init__.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/datamodule/mnist_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/datamodule/mnist_data_module.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/datamodule/tests/test_mnist_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/datamodule/tests/test_mnist_data_module.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/datamodule/tests/test_torchvision_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/datamodule/tests/test_torchvision_data_module.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/datamodule/torchvision_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/datamodule/torchvision_data_module.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/datamodule/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/datamodule/transforms/__init__.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/datamodule/transforms/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/datamodule/transforms/builder.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchrecipes/vision/core/ops/fine_tuning_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/ops/fine_tuning_wrapper.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/optim/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/optim/lr_scheduler.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/tests/test_fine_tuning_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/tests/test_fine_tuning_wrapper.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/tests/test_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/tests/test_lr_scheduler.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/utils/lr_scheduler.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/utils/model_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/utils/model_weights.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/utils/model_weights_exporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/utils/model_weights_exporter.py -------------------------------------------------------------------------------- /torchrecipes/vision/core/utils/test_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/core/utils/test_module.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/README.md -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/__init__.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/callbacks/__init__.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/callbacks/mixup_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/callbacks/mixup_transform.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/__init__.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/datamodule/datasets/cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/datamodule/datasets/cifar10.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/datamodule/datasets/fake_data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/datamodule/datasets/fake_data.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/datamodule/torchvision_datamodule.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/datamodule/torchvision_datamodule.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/default_config.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/default_module.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/module/default_module.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/loss/cross_entropy.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.nn.CrossEntropyLoss 2 | -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/lr_scheduler/step_lr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/module/lr_scheduler/step_lr.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/metrics/accuracy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/module/metrics/accuracy.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/metrics/accuracy_top1_top5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/module/metrics/accuracy_top1_top5.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/metrics/average_precision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/module/metrics/average_precision.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/metrics/multilabel_accuracy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/module/metrics/multilabel_accuracy.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/model/resnet18.yaml: -------------------------------------------------------------------------------- 1 | _target_: torchvision.models.resnet18 2 | pretrained: false 3 | -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/model/resnet50.yaml: -------------------------------------------------------------------------------- 1 | _target_: torchvision.models.resnet50 2 | -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/model/resnext101_32x4d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/module/model/resnext101_32x4d.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/module/optim/sgd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/module/optim/sgd.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/trainer/cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/trainer/cpu.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/conf/trainer/gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/conf/trainer/gpu.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/losses/soft_target_cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/losses/soft_target_cross_entropy_loss.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/losses/tests/test_soft_target_cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/losses/tests/test_soft_target_cross_entropy_loss.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/main.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/metrics/multilabel_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/metrics/multilabel_accuracy.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/metrics/tests/test_multilabel_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/metrics/tests/test_multilabel_accuracy.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/module/image_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/module/image_classification.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/tests/test_image_classification_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/tests/test_image_classification_module.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_classification/tests/test_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_classification/tests/test_main.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/__init__.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/callbacks/__init__.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/callbacks/image_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/callbacks/image_generation.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/__init__.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/datamodule/datasets/fake_data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/datamodule/datasets/fake_data.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/datamodule/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/datamodule/mnist.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/datamodule/torchvision_datamodule.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/datamodule/torchvision_datamodule.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/datamodule/transforms/resize.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/datamodule/transforms/resize.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/gan_train_app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/gan_train_app.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/infogan_train_app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/infogan_train_app.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/criterion/bce_loss.yaml: -------------------------------------------------------------------------------- 1 | _target_: torch.nn.BCELoss 2 | -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/discriminator/dcgan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/module/discriminator/dcgan.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/discriminator/gan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/module/discriminator/gan.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/discriminator/infogan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/module/discriminator/infogan.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/generator/dcgan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/module/generator/dcgan.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/generator/gan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/module/generator/gan.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/generator/infogan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/module/generator/infogan.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/optim/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/module/optim/default.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/optim/infogan_default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/module/optim/infogan_default.yaml -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/conf/module/tests/test_gan_module_conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/conf/module/tests/test_gan_module_conf.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/models/dcgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/models/dcgan.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/models/gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/models/gan.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/models/infogan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/models/infogan.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/module/gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/module/gan.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/module/infogan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/module/infogan.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/tests/test_image_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/tests/test_image_generation.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/tests/test_train_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/tests/test_train_app.py -------------------------------------------------------------------------------- /torchrecipes/vision/image_generation/train_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/recipes/HEAD/torchrecipes/vision/image_generation/train_app.py --------------------------------------------------------------------------------