README.md
pyproject.toml
setup.cfg
setup.py
tests/__init__.py
tests/helpers.py
tests/test_adapters.py
tests/test_callback.py
tests/test_scheduler.py
tests/datamodules/__init__.py
tests/datamodules/helpers.py
tests/datamodules/test_datamodule_cpu.py
tests/datamodules/test_datamodule_ddp.py
tests/datamodules/test_datamodule_dp.py
tests/datamodules/test_datamodule_gpu.py
tests/models/__init__.py
tests/models/helpers.py
tests/models/test_models_cpu.py
tests/models/test_models_ddp.py
tests/models/test_models_dp.py
tests/models/test_models_gpu.py
transformers_lightning/__init__.py
transformers_lightning/info.py
transformers_lightning.egg-info/PKG-INFO
transformers_lightning.egg-info/SOURCES.txt
transformers_lightning.egg-info/dependency_links.txt
transformers_lightning.egg-info/top_level.txt
transformers_lightning/adapters/__init__.py
transformers_lightning/adapters/csv_adapter.py
transformers_lightning/adapters/file_adapter.py
transformers_lightning/adapters/line_adapter.py
transformers_lightning/adapters/super_adapter.py
transformers_lightning/callbacks/__init__.py
transformers_lightning/callbacks/transformers_model_checkpoint.py
transformers_lightning/datamodules/__init__.py
transformers_lightning/datamodules/adapter_datamodule.py
transformers_lightning/datamodules/compressed_datamodule.py
transformers_lightning/datamodules/super_datamodule.py
transformers_lightning/datasets/__init__.py
transformers_lightning/datasets/compressed_dataset.py
transformers_lightning/datasets/iterable_dataset.py
transformers_lightning/datasets/map_dataset.py
transformers_lightning/datasets/stack_dataset.py
transformers_lightning/datasets/super_dataset.py
transformers_lightning/defaults/__init__.py
transformers_lightning/language_modeling/__init__.py
transformers_lightning/language_modeling/language_model.py
transformers_lightning/language_modeling/masked_language_modeling.py
transformers_lightning/language_modeling/random_token_substitution.py
transformers_lightning/language_modeling/utils.py
transformers_lightning/metrics/__init__.py
transformers_lightning/metrics/hit_rate.py
transformers_lightning/models/__init__.py
transformers_lightning/models/transformers_model.py
transformers_lightning/optimizers/__init__.py
transformers_lightning/optimizers/adamw.py
transformers_lightning/optimizers/adamw_electra.py
transformers_lightning/optimizers/super_optimizer.py
transformers_lightning/optimizers/utils.py
transformers_lightning/schedulers/__init__.py
transformers_lightning/schedulers/constant_scheduler.py
transformers_lightning/schedulers/constant_scheduler_with_warmup.py
transformers_lightning/schedulers/cosine_scheduler_with_warmup.py
transformers_lightning/schedulers/cosine_scheduler_with_warmup_and_hard_restart.py
transformers_lightning/schedulers/layerwise_decay_scheduler.py
transformers_lightning/schedulers/linear_scheduler.py
transformers_lightning/schedulers/linear_scheduler_with_warmup.py
transformers_lightning/schedulers/super_scheduler.py
transformers_lightning/utils/__init__.py
transformers_lightning/utils/functional.py
transformers_lightning/utils/generators.py
transformers_lightning/utils/gradient.py
transformers_lightning/utils/inspectors.py
transformers_lightning/utils/language_modeling.py
transformers_lightning/utils/readers.py