Wheelodex » Projects » finetuning-scheduler

finetuning-scheduler

View on PyPI — Reverse Dependencies (0)

2.5.0 finetuning_scheduler-2.5.0-py3-none-any.whl

Wheel Details

Project: finetuning-scheduler
Version: 2.5.0
Filename: finetuning_scheduler-2.5.0-py3-none-any.whl
Download: [link]
Size: 132826
MD5: 4a2396a1fe9929ce2b2f0959f88fea77
SHA256: d7964dec9a4a4f50575c485d9b787650bfdf0fa881de6fd34118fbdebab16401
Uploaded: 2024-12-20 19:13:47 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: finetuning-scheduler
Version: 2.5.0
Summary: A PyTorch Lightning extension that enhances model experimentation with flexible fine-tuning schedules.
Author: Dan Dale
Author-Email: danny.dale[at]gmail.com
Home-Page: https://github.com/speediedan/finetuning-scheduler
Download-Url: https://github.com/speediedan/finetuning-scheduler
Project-Url: Bug Tracker, https://github.com/speediedan/finetuning-scheduler/issues
Project-Url: Documentation, https://finetuning-scheduler.readthedocs.io/en/latest/
Project-Url: Source Code, https://github.com/speediedan/finetuning-scheduler
License: Apache-2.0
Keywords: deep learning,pytorch,AI,machine learning,pytorch-lightning,lightning,fine-tuning,finetuning
Classifier: Environment :: Console
Classifier: Natural Language :: English
Classifier: Development Status :: 5 - Production/Stable
Classifier: Intended Audience :: Developers
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Scientific/Engineering :: Image Recognition
Classifier: Topic :: Scientific/Engineering :: Information Analysis
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.9
Requires-Dist: lightning (<2.5.1,>=2.5.0)
Requires-Dist: torch (>=2.2.0)
Requires-Dist: datasets; extra == "examples"
Requires-Dist: evaluate; extra == "examples"
Requires-Dist: transformers (>=4.18.0); extra == "examples"
Requires-Dist: scikit-learn; extra == "examples"
Requires-Dist: sentencepiece; extra == "examples"
Requires-Dist: tensorboardX (>=2.2); extra == "examples"
Requires-Dist: tabulate; extra == "examples"
Requires-Dist: psutil; extra == "examples"
Requires-Dist: numpy (<2.0); extra == "examples"
Requires-Dist: jsonargparse[signatures] (>=4.27.7); extra == "examples"
Requires-Dist: omegaconf (>=2.1.0); extra == "examples"
Requires-Dist: hydra-core (>=1.1.0); extra == "examples"
Requires-Dist: rich (>=12.3.0); extra == "extra"
Requires-Dist: jsonargparse[signatures] (>=4.27.7); extra == "extra"
Requires-Dist: omegaconf (>=2.1.0); extra == "extra"
Requires-Dist: hydra-core (>=1.1.0); extra == "extra"
Requires-Dist: coverage (>=6.4); extra == "test"
Requires-Dist: pytest (>=6.0); extra == "test"
Requires-Dist: pytest-rerunfailures (>=10.2); extra == "test"
Requires-Dist: twine (==3.2); extra == "test"
Requires-Dist: mypy (>=0.920); extra == "test"
Requires-Dist: pre-commit (>=1.0); extra == "test"
Requires-Dist: protobuf (<=3.20.1); extra == "test"
Requires-Dist: ipython[notebook]; extra == "ipynb"
Requires-Dist: jupytext (>=1.10); extra == "ipynb"
Requires-Dist: nbval (>=0.9.6); extra == "ipynb"
Requires-Dist: jsonargparse[signatures] (>=4.27.7); extra == "cli"
Requires-Dist: omegaconf (>=2.1.0); extra == "cli"
Requires-Dist: hydra-core (>=1.1.0); extra == "cli"
Requires-Dist: rich (>=12.3.0); extra == "dev"
Requires-Dist: jsonargparse[signatures] (>=4.27.7); extra == "dev"
Requires-Dist: omegaconf (>=2.1.0); extra == "dev"
Requires-Dist: hydra-core (>=1.1.0); extra == "dev"
Requires-Dist: coverage (>=6.4); extra == "dev"
Requires-Dist: pytest (>=6.0); extra == "dev"
Requires-Dist: pytest-rerunfailures (>=10.2); extra == "dev"
Requires-Dist: twine (==3.2); extra == "dev"
Requires-Dist: mypy (>=0.920); extra == "dev"
Requires-Dist: pre-commit (>=1.0); extra == "dev"
Requires-Dist: protobuf (<=3.20.1); extra == "dev"
Requires-Dist: ipython[notebook]; extra == "dev"
Requires-Dist: jupytext (>=1.10); extra == "dev"
Requires-Dist: nbval (>=0.9.6); extra == "dev"
Requires-Dist: rich (>=12.3.0); extra == "all"
Requires-Dist: jsonargparse[signatures] (>=4.27.7); extra == "all"
Requires-Dist: omegaconf (>=2.1.0); extra == "all"
Requires-Dist: hydra-core (>=1.1.0); extra == "all"
Requires-Dist: coverage (>=6.4); extra == "all"
Requires-Dist: pytest (>=6.0); extra == "all"
Requires-Dist: pytest-rerunfailures (>=10.2); extra == "all"
Requires-Dist: twine (==3.2); extra == "all"
Requires-Dist: mypy (>=0.920); extra == "all"
Requires-Dist: pre-commit (>=1.0); extra == "all"
Requires-Dist: protobuf (<=3.20.1); extra == "all"
Requires-Dist: ipython[notebook]; extra == "all"
Requires-Dist: jupytext (>=1.10); extra == "all"
Requires-Dist: nbval (>=0.9.6); extra == "all"
Requires-Dist: datasets; extra == "all"
Requires-Dist: evaluate; extra == "all"
Requires-Dist: transformers (>=4.18.0); extra == "all"
Requires-Dist: scikit-learn; extra == "all"
Requires-Dist: sentencepiece; extra == "all"
Requires-Dist: tensorboardX (>=2.2); extra == "all"
Requires-Dist: tabulate; extra == "all"
Requires-Dist: psutil; extra == "all"
Requires-Dist: numpy (<2.0); extra == "all"
Requires-Dist: jsonargparse[signatures] (>=4.27.7); extra == "all"
Requires-Dist: omegaconf (>=2.1.0); extra == "all"
Requires-Dist: hydra-core (>=1.1.0); extra == "all"
Provides-Extra: examples
Provides-Extra: extra
Provides-Extra: test
Provides-Extra: ipynb
Provides-Extra: cli
Provides-Extra: dev
Provides-Extra: all
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 9843 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.6.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
finetuning_scheduler/__about__.py sha256=FNH-PDpM1XH6jaG4qFPm7SyZjDcr9cBIZ1aH1ASD5bQ 2099
finetuning_scheduler/__init__.py sha256=-OQoJvfzZxgwYMPL5oN1pdFt13-uOi_l7KIFp9tar0I 674
finetuning_scheduler/fts.py sha256=G24s_ikypUZ_320JGBNYdIBn-00WArAEpkGIcQdIA_Q 52731
finetuning_scheduler/fts_supporters.py sha256=qaYY-aHW88Cnnu1xYfhub0h-Etc09tMszGF-7FmAiY4 105746
finetuning_scheduler/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
finetuning_scheduler/setup_tools.py sha256=W7N2XWbeHxruiyTz9DDKICDmyVmFrvYqtgkbmNwUiXg 5981
finetuning_scheduler/types.py sha256=81OPgZSyiMhx03O3dxEZiTBLGS_dgzXDUv-z9nwuXko 2025
finetuning_scheduler/strategy_adapters/__init__.py sha256=pTKZ6jfFbebAdaPEvALDpuabF8fo7ti51SX0h4jGFe4 962
finetuning_scheduler/strategy_adapters/_mp_imports.py sha256=kxeeGVuMr3aJQuX4bsHK43htS34W7n02gqHKjbzGbIQ 2020
finetuning_scheduler/strategy_adapters/_wrap_utils.py sha256=yWObtF0q2tdtS7O4as22pJ4LgKwMwu46qDW1RxJF67A 3658
finetuning_scheduler/strategy_adapters/base.py sha256=YcYoKzuggNkSl3aGnrSQSjYImFHFv4uGcd9_v1NE2fA 19873
finetuning_scheduler/strategy_adapters/fsdp.py sha256=a6vgC3KJbeg8X1SVu-v2haNW4BJl2zLRPGLiwvlLbqw 45129
finetuning_scheduler/strategy_adapters/model_parallel.py sha256=Jf9nsr_szLkr2P9abYjysYGe_Xo0PuCEihoMFWqCnYE 24459
fts_examples/__init__.py sha256=4YM5AVZMoyd0iuylZnHLiicjeeKWwiJU7yA1zJ70Xh8 281
fts_examples/cfg_utils.py sha256=LA6CDIsoEkM_TU3Ej-2uM97a62krf4sEonnJ1hC4NPU 2806
fts_examples/cli_experiment_utils.py sha256=RxH2iAI-G4nQyNcVTLp6qnqw_jAx62jWzHMkvt01kNw 14386
fts_examples/fts_superglue.py sha256=rQU39SBDReIV5oQFFbbPmWr8qMHKl0q4x4LSXy9igME 16180
fts_examples/config/RteBoolqModule_ft_schedule_deberta_base.yaml sha256=HUFhO_eI8Ic1RH0j923nH69KkDqvIrem_KvGLYbx_qY 498
fts_examples/config/RteBoolqModule_ft_schedule_deberta_base_fsdp.yaml sha256=3_cL0HapLcpkvm9L9oyNUKKoAOpJJwO82P_HsVzwFUs 650
fts_examples/config/fts_defaults.yaml sha256=ooZ3ocBt-GQsoRs9qOusNRG-Qc7OZmkSttKcZKdN06k 1020
fts_examples/config/fts_explicit.yaml sha256=SXF2pKi30leImB6_ZkNj9Lr-31UjI5sCXmqsXdmeong 676
fts_examples/config/fts_implicit.yaml sha256=d7APYLzKyBngjMNiz-eXZOJ5BJbCZXttwe9EE3xAq8M 559
fts_examples/config/nofts_baseline.yaml sha256=9aRZvObnj7KiaLEDdpCyrWD08cZeqKYNdsQd9wNLeVI 340
fts_examples/config/advanced/fsdp/fts_ddp_fsdp_baseline_profile.yaml sha256=L2u92P5YTAyBUNpjUpccwyPK45C_tcUFjx62zFHpWWM 1248
fts_examples/config/advanced/fsdp/fts_fsdp_awp_overrides_offload_profile.yaml sha256=NzubjFQSNfiWXZ6OpFgDIf5LCXa2Abd4e4bZeNoOZ6s 2151
fts_examples/config/advanced/fsdp/fts_fsdp_awp_overrides_profile.yaml sha256=5ci5z18nJ6JZu4wpiZ1DBtcuTnrrxSzJTvElws42Jtc 2352
fts_examples/config/advanced/fsdp/fts_fsdp_basic_profile.yaml sha256=jMd5bVKCE6F_HoLbAs0w_pyBhgWl16-62ENATpRxe84 1914
fts_examples/config/advanced/reinit_lr/explicit_reinit_lr.yaml sha256=9Ki3gn_0msWY3QBf-ERt2cT0V6cOzyh54-zi5zLZzDg 1136
fts_examples/config/advanced/reinit_lr/fts_explicit_reinit_lr.yaml sha256=KD6mNeFxZZhUAOyalqm14OL8UwvQZeCK-3uR25syPjA 1138
fts_examples/config/advanced/reinit_lr/fts_implicit_reinit_lr.yaml sha256=W0_pLLKxqxLXU-CitPhFlPPUlHHJC4K-NRl_AHhJ824 1517
fts_examples/config/advanced/reinit_optim_lr/explicit_reinit_optim_lr.yaml sha256=3OxkKXEOmnh4av28b8Hr07ncRMFnIRCIbmTJ0Ei9fAs 1457
fts_examples/config/advanced/reinit_optim_lr/fts_explicit_reinit_optim_lr.yaml sha256=HVvl6qGQ05tziGIM5Q6h56Y7N1t1jZsl9cWf9XWdU00 1235
fts_examples/config/advanced/reinit_optim_lr/fts_implicit_reinit_optim_lr.yaml sha256=WMup9M06IMVJWlHboaAP3SY8qzrEKfpj8V9kKbPv-xI 1738
fts_examples/config/advanced/reinit_optim_lr/fts_implicit_reinit_optim_lr_use_curr.yaml sha256=dSZsP8y0thX8C_SEcTZ3XfFd5TyMNnqBDfljSB3xaok 1781
fts_examples/model_parallel/mp_examples.py sha256=5zJX-lQxcarWrEGGOF8pdBr86v6CSTe9dKpmDCpX0cI 5371
fts_examples/model_parallel/torchtitan_llama.py sha256=8XBQzDC68T-6_rI9txyRvfBUQCDMpHqbEZInRszaUt8 18181
fts_examples/model_parallel/config/fts_fsdp_auto_plan.yaml sha256=z3R0YanpuG0QDz2MwKYMhaq7T2TEEFKaKbwrYcS0_rY 687
fts_examples/model_parallel/config/fts_fsdp_profiling.yaml sha256=NDvX2xkklFv6UBbz5E-kB3r5lzoAGFr1jgTWJXN82Gs 534
fts_examples/model_parallel/config/fts_tp_plan.yaml sha256=khTaawiWCSX1RMt-OfT4yfQdvf8vpWmhqw9egvaSAaQ 324
fts_examples/model_parallel/config/defaults/fts_mp_example_defaults.yaml sha256=h7vLWbNuWg4y6n1eKBhTg2N7JM-SdsafCuMWIahpb18 1830
fts_examples/model_parallel/config/defaults/llama_ft_schedule.yaml sha256=lDHMOgan5WRJBpzGU4RcH2fE8ZOtCalIYWwBQiAvBO0 316
fts_examples/model_parallel/config/defaults/torchtitan_llama.yaml sha256=CAu9yMu3Z9nkrkEaEpHjpVRKqbRaK45fgyWKT7dBUT0 84
fts_examples/model_parallel/config/profiling/memprofiler_demo.yaml sha256=-n-x4m2A1mW9SMrczf0ZRo5BVQ02NDa_zcP4zWtAp0A 1384
fts_examples/model_parallel/config/profiling/pytorch_profiler.yaml sha256=p_QQdhXObweFicIBQgLndQZA_5G9xUb1NbPmIC9DsZo 477
fts_examples/patching/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
fts_examples/patching/_patch_utils.py sha256=AKeHj0kMWCL0X1AOpJFxo1VcTW7vY1pVoyg6rgcZwQc 1057
fts_examples/patching/dep_patch_shim.py sha256=tTL8sCICtDduTf0N-lQQw2CAst0syrNdgzXk0sepQOo 5571
fts_examples/patching/patched_einsum_strategies.py sha256=PayivbRSCohaicGBDOU6Uain96zdR6F8jSrHrHdHIto 4438
fts_examples/patching/patched_lightning_jsonargparse.py sha256=lIharAHycok1MEMC_z3cw_fpQ2IJGd8WjAzUb_w8rfk 764
fts_examples/patching/patched_numpyarrowextractor.py sha256=xVYOR1ZL7cxCWOD1Xo8uYSkp_NKJuHPG2c-8JmvcnTo 2950
fts_examples/patching/patched_triton_jit_fn_init.py sha256=fZAHxfzUcm3cI04Hqd_6e_1h81XPrqhV1oUd4CilikE 2230
fts_examples/profiling/__init__.py sha256=cC_6JpqFJz5of7m61hBcRtDANg7safMVcxFBRcBc-bQ 699
fts_examples/profiling/config.py sha256=oqzfCLC1g0N3n5L9XLqzX5zcfZACkjZNYOlQxgeCCZA 7141
fts_examples/profiling/extended_profiler.py sha256=bwwJvltsBpqR52Rb1Nobpp-6kyp2W7_AGpvYRaYgeqQ 4854
fts_examples/profiling/memprofiler.py sha256=nPOAG9-PbRR3UTkBm1xkgZQnC2ptwS2cu7WQymQYxFM 28302
fts_examples/profiling/npp_hooks.py sha256=E27C0y1S01NftINrG1tJiUfuqtQHxJJRWaFcLyr1swk 1145
fts_examples/profiling/profiler_hooks_mixin.py sha256=E4Yq77k2U_VO7IDarGmS0lSNkfc_cTnT0bBk1TmtRgA 1799
finetuning_scheduler-2.5.0.dist-info/LICENSE sha256=pB4vrpkVrlYCj429C_J5lfGQdhPLzS2BphsBCtNOn-k 11349
finetuning_scheduler-2.5.0.dist-info/METADATA sha256=G4Dh7ozxKwnEKNxCm7GZhVVxmXOvV8q40rTZmhx2OKg 14969
finetuning_scheduler-2.5.0.dist-info/WHEEL sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U 91
finetuning_scheduler-2.5.0.dist-info/top_level.txt sha256=QhoxAdvNZVW5Q_0l9Nrg91ILeo803XoX5wpzqXm2bb4 34
finetuning_scheduler-2.5.0.dist-info/RECORD — —

top_level.txt

finetuning_scheduler
fts_examples