bert-squeeze

View on PyPIReverse Dependencies (0)

0.1.1 bert_squeeze-0.1.1-py3-none-any.whl

Wheel Details

Project: bert-squeeze
Version: 0.1.1
Filename: bert_squeeze-0.1.1-py3-none-any.whl
Download: [link]
Size: 92945
MD5: 0d94a6c7027fff6506301f768f60968c
SHA256: 20b75fe6d291e695f3aeff2f9dbe731d0593428443aebd61cc922ab930f0923a
Uploaded: 2023-06-16 09:10:14 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: bert-squeeze
Version: 0.1.1
Summary: Tools for Transformers compression using PyTorch Lightning
Author: JulesBelveze
Author-Email: jules.belveze[at]hotmail.fr
Keywords: nlp,transformers,bert
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Text Processing
Requires-Python: >=3.8.1,<4.0
Requires-Dist: datasets (<2.0.0,>=1.8.0)
Requires-Dist: huggingface-hub (==0.13.4)
Requires-Dist: hydra-core (<2.0.0,>=1.1.1)
Requires-Dist: lightning (<3.0.0,>=2.0.0)
Requires-Dist: matplotlib (<3.7.1)
Requires-Dist: neptune-client[pytorch-lightning] (<0.15.0,>=0.14.0)
Requires-Dist: numpy (<1.20.0)
Requires-Dist: omegaconf (<3.0.0,>=2.1.0)
Requires-Dist: overrides (<7.0.0,>=6.1.0)
Requires-Dist: psutil (<6.0.0,>=5.8.0)
Requires-Dist: python-dotenv (<0.19.0,>=0.18.0)
Requires-Dist: scikit-learn (<2.0.0,>=1.2.2)
Requires-Dist: seaborn (<0.13.0,>=0.12.2)
Requires-Dist: tabulate (<0.9.0,>=0.8.9)
Requires-Dist: torch (<3.0.0,>=2.0.1)
Requires-Dist: transformers (<5.0.0,>=4.20.0)
[No description]

WHEEL

Wheel-Version: 1.0
Generator: poetry-core 1.4.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
bert_squeeze/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
bert_squeeze/assistants/__init__.py sha256=0CKlVLIz6Cd2Z2iKY2F6p4kBuNR2NIA_usgEFe0D_X4 90
bert_squeeze/assistants/configs/distil.yaml sha256=Mw-kaCCMcLWpoRBF6FJHkb1RSEx1jWwFL7ZPQklm2Hg 1953
bert_squeeze/assistants/configs/distil_hard.yaml sha256=sljqb6TfaBjcTcroqiyJevNOqJggs1awhPzdpg8aQCU 2251
bert_squeeze/assistants/configs/distil_parallel.yaml sha256=sz-Bkq5pelxMipqzOT-Mos_1sS5iBFcrVwEYfyjDpf8 1767
bert_squeeze/assistants/configs/distil_soft.yaml sha256=8tacjtrGYkK3NXzPGMrDJaYXPFut8BnUtj0zVDiTCNo 2094
bert_squeeze/assistants/configs/train_bert.yaml sha256=eJW9Nr-VLWS2gghvJ0XT7A6lgYXiQN0xA3lxILwY1pM 978
bert_squeeze/assistants/configs/train_deebert.yaml sha256=0mNX7AiVvajwx4LQ74qrsr-md5HBB0VAouZZbfQIM30 1061
bert_squeeze/assistants/configs/train_fastbert.yaml sha256=I6jm5BI4xNgijU7w2Fo4hbDsflPNMC_Se1RynpHh-4Q 1085
bert_squeeze/assistants/configs/train_lr.yaml sha256=qnWJV0EkPjkyCHvGkOrMyee0yaHZBZfrOz4yEB3MSQg 852
bert_squeeze/assistants/configs/train_lstm.yaml sha256=ZYf1C3OwqGyO87FQ7YqfYboLASlI-z1nITtVdua_1R0 774
bert_squeeze/assistants/configs/train_theseus_bert.yaml sha256=c9u4jYfo9UQm6zKVe2MQ1HBU69vlQs6u6MRzEMdM-jA 1084
bert_squeeze/assistants/distil_assistant.py sha256=6EqzWekzD7TRkStSz1Y3ckZDhNqPJN1JaxiKhu7Jypo 7570
bert_squeeze/assistants/train_assistant.py sha256=_u_MQozQlKBpoyZ8PL_3kb9AmTbhhNfRrNNWgbYOBAs 5756
bert_squeeze/data/__init__.py sha256=8UVpJ8f-ZVOpd1Pq03fETQ5Qsslt-bVvDx7o02sImqU 145
bert_squeeze/data/local_datasets/__init__.py sha256=cKSzGx-VAUJBxPUIHsXvG7ZVkaVr9y8fOPywWmqD4Ps 98
bert_squeeze/data/local_datasets/conference_dataset.py sha256=I2o1n1CPfTS101iFHboR7Y3qVEW6_jFiytLFCiKqUec 2471
bert_squeeze/data/local_datasets/parallel_dataset.py sha256=osw1Zv-Ypw1qsaKYy_b7Bux92pZtFDO_zIHAr4axil4 2647
bert_squeeze/data/local_datasets/unlabeled_dataset.py sha256=ujZ92qJVj3T6xnznqio7MaPsIWT9r8w3MN4-U-EjgvI 1915
bert_squeeze/data/modules/__init__.py sha256=qZR2b4jsGslTG8kMdjQ8OuA-1bxrMS89SEbbpU9gETg 186
bert_squeeze/data/modules/distillation_module.py sha256=Csz71S5BbHyvcka47Q3V-TTqtT19HMm1_537bv3r84o 8869
bert_squeeze/data/modules/lr_module.py sha256=bpu4310v6ZCPJNmvQv0oddp3C5JYtwp_EW7LQk_AOhU 4552
bert_squeeze/data/modules/lstm_module.py sha256=XiYC92U95q6sXjFD-wn6szpiDJm2f6Zrx-M4vn5yphk 6397
bert_squeeze/data/modules/transformer_module.py sha256=WpAg1izB92P_55AH_FpR3gSzYp79Bau2dguGb7NUMVM 6073
bert_squeeze/distillation/__init__.py sha256=Jl00ZJSJClD4cwCosBcn3Ik9cMXGQPAy1ug_eo0A4MY 33
bert_squeeze/distillation/distiller.py sha256=mgBo6lfWSftlz3mxbAI_NTPcVMc99Mmk_o01n2RTtqE 24364
bert_squeeze/distillation/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
bert_squeeze/distillation/utils/labeler.py sha256=VBE5cMQWrN8z5LiXY5L3NWehVa2spggLMpBwIEKTAM8 5418
bert_squeeze/inference/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
bert_squeeze/inference/model.py sha256=8EaT4_cyZysGETGIboK2kqJ22ysMETXO85r8eG10bK0 1960
bert_squeeze/inference/processors.py sha256=pAxFY7pyP_Z5KruV0q3WYPeB4A4xp9OJqskqzeyq1jw 692
bert_squeeze/models/__init__.py sha256=-rb0CGsEx9Ekhi6nFNeXabQ34Kj4CmGgqsHjcoQK_AY 256
bert_squeeze/models/base_lt_module.py sha256=3XPSeaAaHToo6AoE2cYxIdsrl4aICIV57BmU69pI9EQ 13173
bert_squeeze/models/custom_transformers/__init__.py sha256=C5W8ZgSADgP7SjwTOtZXPjQFrHSRRqavsDPiJKt6SBY 113
bert_squeeze/models/custom_transformers/bert.py sha256=miblGQDXcKhVwBzp4pkIOhG_D09h-bg35PdaBCXibic 8111
bert_squeeze/models/custom_transformers/deebert.py sha256=G0m8-w50RfzWfwa2X6ch8UHRYGLrsPro6GfngEk21LY 13283
bert_squeeze/models/custom_transformers/fastbert.py sha256=xcohEycQauzRkir5ni5-ZwCDC-PLv75rYh-aXnkvaFM 8452
bert_squeeze/models/custom_transformers/theseus_bert.py sha256=-HBO5tVDwHNb0VTZuvTRd5m4vs09hlDUIEsQrYAZOLw 13974
bert_squeeze/models/layers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
bert_squeeze/models/layers/classifier.py sha256=kvoGqtFt0yDZfSyLx97IomG4-o1axe7kmuRsB2UmobM 1387
bert_squeeze/models/layers/mha.py sha256=j1KdWcYFc8nVlXt2dxk7JYmoHY4XCWvDfGgAEWt7uxk 1993
bert_squeeze/models/lr.py sha256=Tm4OggoECf1B3Jk5UowYZ6X5_hklApFHUtA_OUMrgkU 7219
bert_squeeze/models/lstm.py sha256=-p0I9H_3qQeF56CFjeEG-QMFELj35iCW8cnJvgm_7fs 8435
bert_squeeze/models/lt_bert.py sha256=HkhSqoqB7Rb_FRbMkdRlX2bQTaHJ1OrZjK6jVnKpIz0 5422
bert_squeeze/models/lt_deebert.py sha256=LPvJ8AFMrIKqNE0SmmDxQIeDjE0k9uQ4FIbYhzLjpms 14299
bert_squeeze/models/lt_distilbert.py sha256=l8bpawkhZx-E9suiwVzNFXWWBXB39N1p56UcvkjMUyQ 4529
bert_squeeze/models/lt_fastbert.py sha256=zlXPEg2qu9Wi-dng_47cXM85okZvPEIxMJrwW4bIjDw 11312
bert_squeeze/models/lt_theseus_bert.py sha256=fwPXFUquf2ndhfQkneNYBZ0OT5Izn6J2yD05Iw3FnqI 6007
bert_squeeze/utils/__init__.py sha256=Wi_2MXzi5qNuoMHeCutsqKhCLCzRd7GsF9jk29pfQlY 182
bert_squeeze/utils/artifacts/__init__.py sha256=PAKcZxRQuPFNHl3iJaL1i0Ov3V8wR3nKE8whtMOJrbI 172
bert_squeeze/utils/artifacts/distillation_artifacts.py sha256=lN4RVFPVxaG4dLnHDnSo85G08pIPtFVVKtJkrcX3ST8 2356
bert_squeeze/utils/artifacts/transformer_artifacts.py sha256=Cx6e2VeGDri3dmYYhOs9ozx-wtDu9uOO_K65j2yRmjk 759
bert_squeeze/utils/callbacks/__init__.py sha256=oKfql6lNtrVfBziq6I5UNIVLNNdD_-t4gw0kBWRgdQM 95
bert_squeeze/utils/callbacks/checkpointing.py sha256=YrBpeRhEvHgjpmu-K88N2Bqd0fHsctGE708jAghmE28 1651
bert_squeeze/utils/callbacks/fastbert_logic.py sha256=peDzwQ_2bftXtRCl308Q_IZbvvxrTBaJqBte1af4mCU 1527
bert_squeeze/utils/callbacks/lottery_ticket.py sha256=MBrp_ij5qNt381vtlpmPzO1z77N8WKHez2nq0HZRaWU 14219
bert_squeeze/utils/callbacks/pruning.py sha256=z52wtZYPcmgl-0IKB6l47w9g01rZrGm2294zMDpPKsQ 5984
bert_squeeze/utils/callbacks/quantization.py sha256=PJKf9mu-mB8x-6o8aLmJmmVeXmNH6V5_LtZJlIKstbo 1420
bert_squeeze/utils/errors/__init__.py sha256=FFZiTEVyTeAwR1lIMRodhuukMYGRebNhxFKtaYyzhBo 92
bert_squeeze/utils/errors/config_errors.py sha256=k76NeTK6zwvuQ9BgpXcCC9i0OjJ6WjHEj5rxuOn-Jd8 167
bert_squeeze/utils/errors/deebert_errors.py sha256=EGo4476-F12CaDP6yIN1j44FzEM7srGNdWQiuMPIkdI 418
bert_squeeze/utils/losses/__init__.py sha256=ku6WrdT3fZjVo8JFxtr0IGsItPNWLJkXYQwMA7VTxbs 171
bert_squeeze/utils/losses/distillation_losses.py sha256=vSonuIaihIzEmX5kHkvRbfeEajRZiwLlaHv-HdvGy8o 1226
bert_squeeze/utils/losses/losses.py sha256=ncYbn2fAA1ZnZRk4WHvHJhe3v6hsj-eRogP1SYpyITw 1026
bert_squeeze/utils/losses/lsl.py sha256=jk2IkL_WiSJ7Y46QkwTV2wPr-TNPnbzY0vtCOjEncVQ 1631
bert_squeeze/utils/losses/romebert_loss.py sha256=d2WQ8VMX0yIOS8s-v3KgeuvL95VAxxz4tPUdXFnA494 3102
bert_squeeze/utils/optimizers/__init__.py sha256=x2y9GMDQ_XwEcnWM1O1hikkCAcAUY7w_4uKVolOotuY 32
bert_squeeze/utils/optimizers/bert_adam.py sha256=t8Iijcy9ELUVMiuXBl48wYvSumxrm1JMdyNmSfqUsq8 7979
bert_squeeze/utils/schedulers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
bert_squeeze/utils/schedulers/theseus_schedulers.py sha256=_xppq0JM6x72AwC1LxfWNWZfmN18IkKR-6kY-YKyoq4 3387
bert_squeeze/utils/scorers/__init__.py sha256=CxdkNz4Y6UsQ_5ggQW_nXM41le0vkkdGSBWM_X7WJOM 80
bert_squeeze/utils/scorers/scorer.py sha256=pg8mRAdScMWD0tfc3lsrpI3u6PU7lSD-euT_yi5tOMw 10543
bert_squeeze/utils/scorers/scorer_fast.py sha256=959iAaxy98PfKGasTXV_l0Us2_1-HAMIqYFY0bh-BaI 6941
bert_squeeze/utils/types.py sha256=JlUORMcS4alXp6rkqDJmHFPt_blaWAlGYnd1k1UF2Xc 1296
bert_squeeze/utils/utils_fct.py sha256=15vkjtPJc8dxhAef5iJwv0pHBflVvl4pm2Zy3ph5gWw 3053
bert_squeeze/utils/vocabulary.py sha256=XPhOE0-v-fkueHy65s1lM0mYdanWTn-Pl4qeMmx7-T0 2493
bert_squeeze-0.1.1.dist-info/METADATA sha256=yAu4gBGmEJXxljRK90nNQeevA_VxW91uUj0XYMc0oqI 1239
bert_squeeze-0.1.1.dist-info/WHEEL sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY 88
bert_squeeze-0.1.1.dist-info/RECORD