├── .gitignore ├── CITATION.cff ├── LICENSE ├── README.md ├── cfgs ├── audioset │ ├── efficientnet-b0-leaf-default.cfg │ ├── efficientnet-b0-leaf-default_gcs.cfg │ └── resnet-18-leaf-default.cfg ├── speechcommands │ ├── efficientnet-b0-leaf-default.cfg │ ├── efficientnet-b0-leaf-default_gcs.cfg │ ├── efficientnet-b0-leaf-default_kaiming.cfg │ ├── efficientnet-b0-leaf-default_packed.cfg │ ├── efficientnet-b0-leaf-default_random.cfg │ ├── efficientnet-b0-leaf-default_xavier.cfg │ ├── resnet-18-leaf-default.cfg │ ├── resnet-18-leaf-default_kaiming.cfg │ └── resnet-18-leaf-default_xavier.cfg └── voxceleb │ ├── efficientnet-b0-leaf-default.cfg │ └── resnet-18-leaf-default.cfg ├── leaf_pytorch ├── __init__.py ├── convolution.py ├── filters.py ├── frontend.py ├── frontend_helper.py ├── impulse_responses.py ├── initializers.py ├── pooling.py ├── postprocessing.py └── utils.py ├── models ├── __init__.py ├── classifier.py ├── densenet.py ├── efficientnet │ ├── __init__.py │ ├── model.py │ └── utils.py ├── model_helper.py ├── netvlad.py ├── resnet.py └── vanilla_cifar_resnet.py ├── speechcommands_init_exps.sh ├── test.py ├── test_leaf.py ├── train.py ├── train_xla.py ├── utilities ├── __init__.py ├── config_parser.py ├── data │ ├── __init__.py │ ├── mixup.py │ ├── packed_dataset.py │ ├── raw_dataset.py │ ├── raw_transforms.py │ ├── raw_waveform_parser.py │ └── utils.py ├── metrics_helper.py └── training_utils.py └── voxceleb_run.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/audioset/efficientnet-b0-leaf-default.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/audioset/efficientnet-b0-leaf-default.cfg -------------------------------------------------------------------------------- /cfgs/audioset/efficientnet-b0-leaf-default_gcs.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/audioset/efficientnet-b0-leaf-default_gcs.cfg -------------------------------------------------------------------------------- /cfgs/audioset/resnet-18-leaf-default.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/audioset/resnet-18-leaf-default.cfg -------------------------------------------------------------------------------- /cfgs/speechcommands/efficientnet-b0-leaf-default.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/speechcommands/efficientnet-b0-leaf-default.cfg -------------------------------------------------------------------------------- /cfgs/speechcommands/efficientnet-b0-leaf-default_gcs.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/speechcommands/efficientnet-b0-leaf-default_gcs.cfg -------------------------------------------------------------------------------- /cfgs/speechcommands/efficientnet-b0-leaf-default_kaiming.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/speechcommands/efficientnet-b0-leaf-default_kaiming.cfg -------------------------------------------------------------------------------- /cfgs/speechcommands/efficientnet-b0-leaf-default_packed.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/speechcommands/efficientnet-b0-leaf-default_packed.cfg -------------------------------------------------------------------------------- /cfgs/speechcommands/efficientnet-b0-leaf-default_random.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/speechcommands/efficientnet-b0-leaf-default_random.cfg -------------------------------------------------------------------------------- /cfgs/speechcommands/efficientnet-b0-leaf-default_xavier.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/speechcommands/efficientnet-b0-leaf-default_xavier.cfg -------------------------------------------------------------------------------- /cfgs/speechcommands/resnet-18-leaf-default.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/speechcommands/resnet-18-leaf-default.cfg -------------------------------------------------------------------------------- /cfgs/speechcommands/resnet-18-leaf-default_kaiming.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/speechcommands/resnet-18-leaf-default_kaiming.cfg -------------------------------------------------------------------------------- /cfgs/speechcommands/resnet-18-leaf-default_xavier.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/speechcommands/resnet-18-leaf-default_xavier.cfg -------------------------------------------------------------------------------- /cfgs/voxceleb/efficientnet-b0-leaf-default.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/voxceleb/efficientnet-b0-leaf-default.cfg -------------------------------------------------------------------------------- /cfgs/voxceleb/resnet-18-leaf-default.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/cfgs/voxceleb/resnet-18-leaf-default.cfg -------------------------------------------------------------------------------- /leaf_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/__init__.py -------------------------------------------------------------------------------- /leaf_pytorch/convolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/convolution.py -------------------------------------------------------------------------------- /leaf_pytorch/filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/filters.py -------------------------------------------------------------------------------- /leaf_pytorch/frontend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/frontend.py -------------------------------------------------------------------------------- /leaf_pytorch/frontend_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/frontend_helper.py -------------------------------------------------------------------------------- /leaf_pytorch/impulse_responses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/impulse_responses.py -------------------------------------------------------------------------------- /leaf_pytorch/initializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/initializers.py -------------------------------------------------------------------------------- /leaf_pytorch/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/pooling.py -------------------------------------------------------------------------------- /leaf_pytorch/postprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/postprocessing.py -------------------------------------------------------------------------------- /leaf_pytorch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/leaf_pytorch/utils.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/models/classifier.py -------------------------------------------------------------------------------- /models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/models/densenet.py -------------------------------------------------------------------------------- /models/efficientnet/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import EfficientNet -------------------------------------------------------------------------------- /models/efficientnet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/models/efficientnet/model.py -------------------------------------------------------------------------------- /models/efficientnet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/models/efficientnet/utils.py -------------------------------------------------------------------------------- /models/model_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/models/model_helper.py -------------------------------------------------------------------------------- /models/netvlad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/models/netvlad.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/vanilla_cifar_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/models/vanilla_cifar_resnet.py -------------------------------------------------------------------------------- /speechcommands_init_exps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/speechcommands_init_exps.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/test.py -------------------------------------------------------------------------------- /test_leaf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/test_leaf.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/train.py -------------------------------------------------------------------------------- /train_xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/train_xla.py -------------------------------------------------------------------------------- /utilities/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utilities/config_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/utilities/config_parser.py -------------------------------------------------------------------------------- /utilities/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utilities/data/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/utilities/data/mixup.py -------------------------------------------------------------------------------- /utilities/data/packed_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/utilities/data/packed_dataset.py -------------------------------------------------------------------------------- /utilities/data/raw_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/utilities/data/raw_dataset.py -------------------------------------------------------------------------------- /utilities/data/raw_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/utilities/data/raw_transforms.py -------------------------------------------------------------------------------- /utilities/data/raw_waveform_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/utilities/data/raw_waveform_parser.py -------------------------------------------------------------------------------- /utilities/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/utilities/data/utils.py -------------------------------------------------------------------------------- /utilities/metrics_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/utilities/metrics_helper.py -------------------------------------------------------------------------------- /utilities/training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/utilities/training_utils.py -------------------------------------------------------------------------------- /voxceleb_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SarthakYadav/leaf-pytorch/HEAD/voxceleb_run.sh --------------------------------------------------------------------------------