├── .flake8 ├── .github └── workflows │ ├── main.yml │ └── pip.yml ├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── Makefile ├── README.md ├── docs └── logo.png ├── fluence ├── __init__.py ├── adaptive │ ├── __init__.py │ ├── entmax.py │ ├── layerdrop.py │ └── span.py ├── datasets │ ├── __init__.py │ └── meta_dataset.py ├── models │ ├── __init__.py │ ├── orthogonal.py │ └── siamese_model.py ├── optim │ ├── __init__.py │ ├── lamb.py │ └── lookahead.py ├── pooling.py ├── prune.py └── sampling │ ├── __init__.py │ └── clustering.py ├── setup.py └── tests ├── fixtures ├── models │ ├── __init__.py │ ├── cbow.py │ └── lenet.py └── tests_samples │ └── MRPC │ ├── cached_train_AlbertTokenizer_128_mrpc │ ├── cached_train_AlbertTokenizer_128_mrpc.lock │ ├── dev.tsv │ └── train.tsv ├── test_adaptive.py ├── test_clustering.py ├── test_datasets.py ├── test_models.py ├── test_optim.py ├── test_pooling.py └── test_prune.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.github/workflows/pip.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/.github/workflows/pip.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/README.md -------------------------------------------------------------------------------- /docs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/docs/logo.png -------------------------------------------------------------------------------- /fluence/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.1.7" 2 | -------------------------------------------------------------------------------- /fluence/adaptive/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/adaptive/__init__.py -------------------------------------------------------------------------------- /fluence/adaptive/entmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/adaptive/entmax.py -------------------------------------------------------------------------------- /fluence/adaptive/layerdrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/adaptive/layerdrop.py -------------------------------------------------------------------------------- /fluence/adaptive/span.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/adaptive/span.py -------------------------------------------------------------------------------- /fluence/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/datasets/__init__.py -------------------------------------------------------------------------------- /fluence/datasets/meta_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/datasets/meta_dataset.py -------------------------------------------------------------------------------- /fluence/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/models/__init__.py -------------------------------------------------------------------------------- /fluence/models/orthogonal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/models/orthogonal.py -------------------------------------------------------------------------------- /fluence/models/siamese_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/models/siamese_model.py -------------------------------------------------------------------------------- /fluence/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/optim/__init__.py -------------------------------------------------------------------------------- /fluence/optim/lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/optim/lamb.py -------------------------------------------------------------------------------- /fluence/optim/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/optim/lookahead.py -------------------------------------------------------------------------------- /fluence/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/pooling.py -------------------------------------------------------------------------------- /fluence/prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/prune.py -------------------------------------------------------------------------------- /fluence/sampling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/sampling/__init__.py -------------------------------------------------------------------------------- /fluence/sampling/clustering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/fluence/sampling/clustering.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/setup.py -------------------------------------------------------------------------------- /tests/fixtures/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/fixtures/models/__init__.py -------------------------------------------------------------------------------- /tests/fixtures/models/cbow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/fixtures/models/cbow.py -------------------------------------------------------------------------------- /tests/fixtures/models/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/fixtures/models/lenet.py -------------------------------------------------------------------------------- /tests/fixtures/tests_samples/MRPC/cached_train_AlbertTokenizer_128_mrpc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/fixtures/tests_samples/MRPC/cached_train_AlbertTokenizer_128_mrpc -------------------------------------------------------------------------------- /tests/fixtures/tests_samples/MRPC/cached_train_AlbertTokenizer_128_mrpc.lock: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/fixtures/tests_samples/MRPC/dev.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/fixtures/tests_samples/MRPC/dev.tsv -------------------------------------------------------------------------------- /tests/fixtures/tests_samples/MRPC/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/fixtures/tests_samples/MRPC/train.tsv -------------------------------------------------------------------------------- /tests/test_adaptive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/test_adaptive.py -------------------------------------------------------------------------------- /tests/test_clustering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/test_clustering.py -------------------------------------------------------------------------------- /tests/test_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/test_datasets.py -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /tests/test_optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/test_optim.py -------------------------------------------------------------------------------- /tests/test_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/test_pooling.py -------------------------------------------------------------------------------- /tests/test_prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prajjwal1/fluence/HEAD/tests/test_prune.py --------------------------------------------------------------------------------