├── .gitignore ├── LICENSE.md ├── README.md ├── docs ├── architecture-organization.png ├── config.png ├── dashboard.png ├── dataset-info.json ├── feed-forward.png ├── lex-glue-data-module.png ├── metrics.png ├── old-README.md ├── the-steps.png ├── train.png ├── transformer-constructor.png ├── transformer-structure.png └── unfair-tos-diagram.png ├── environment.yml └── lightning_mlflow ├── architectures ├── __init__.py └── fine_tune_clsify_head.py ├── config ├── __init__.py └── train_config.py ├── data ├── __init__.py └── lex_glue.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/README.md -------------------------------------------------------------------------------- /docs/architecture-organization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/architecture-organization.png -------------------------------------------------------------------------------- /docs/config.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/config.png -------------------------------------------------------------------------------- /docs/dashboard.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/dashboard.png -------------------------------------------------------------------------------- /docs/dataset-info.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/dataset-info.json -------------------------------------------------------------------------------- /docs/feed-forward.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/feed-forward.png -------------------------------------------------------------------------------- /docs/lex-glue-data-module.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/lex-glue-data-module.png -------------------------------------------------------------------------------- /docs/metrics.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/metrics.png -------------------------------------------------------------------------------- /docs/old-README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/old-README.md -------------------------------------------------------------------------------- /docs/the-steps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/the-steps.png -------------------------------------------------------------------------------- /docs/train.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/train.png -------------------------------------------------------------------------------- /docs/transformer-constructor.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/transformer-constructor.png -------------------------------------------------------------------------------- /docs/transformer-structure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/transformer-structure.png -------------------------------------------------------------------------------- /docs/unfair-tos-diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/docs/unfair-tos-diagram.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/environment.yml -------------------------------------------------------------------------------- /lightning_mlflow/architectures/__init__.py: -------------------------------------------------------------------------------- 1 | from .fine_tune_clsify_head import TransformerModule 2 | -------------------------------------------------------------------------------- /lightning_mlflow/architectures/fine_tune_clsify_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/lightning_mlflow/architectures/fine_tune_clsify_head.py -------------------------------------------------------------------------------- /lightning_mlflow/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/lightning_mlflow/config/__init__.py -------------------------------------------------------------------------------- /lightning_mlflow/config/train_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/lightning_mlflow/config/train_config.py -------------------------------------------------------------------------------- /lightning_mlflow/data/__init__.py: -------------------------------------------------------------------------------- 1 | from .lex_glue import LexGlueDataModule 2 | -------------------------------------------------------------------------------- /lightning_mlflow/data/lex_glue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/lightning_mlflow/data/lex_glue.py -------------------------------------------------------------------------------- /lightning_mlflow/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjohn77/lightning-mlflow-hf/HEAD/lightning_mlflow/train.py --------------------------------------------------------------------------------