├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── examples ├── _pre.py ├── _pre_cv.py ├── _pre_ms.py ├── _pre_nlp.py ├── ae.py ├── cl.py ├── cl_ddp.py ├── cl_ddp_spawn.py ├── cv.py ├── cv_ddp.py ├── cv_ddp_spawn.py ├── dqn.py ├── gan.py ├── gnn_edge.py ├── gnn_graph.py ├── gnn_node.py ├── meta_learning.py ├── nlp_baichuan_sft_lora.py ├── nlp_bert_mlm.py ├── nlp_bert_seq_cls.py ├── nlp_chatglm2_sft_lora.py ├── nlp_gpt_lm.py ├── nlp_gpt_seq_cls.py ├── nlp_gpt_zh_sft_adapter.py ├── nlp_gpt_zh_sft_lora.py ├── test_env.py └── vae.py ├── mini_lightning ├── __init__.py ├── _mini_lightning.py ├── _types.py ├── _utils.py ├── _visualize.py └── _warmup_lrs.py ├── requirements.txt ├── setup.py └── tests ├── test_mini_lightning.py ├── test_select_device.py ├── test_url.py ├── test_utils.py └── test_visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | include requirements.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/README.md -------------------------------------------------------------------------------- /examples/_pre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/_pre.py -------------------------------------------------------------------------------- /examples/_pre_cv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/_pre_cv.py -------------------------------------------------------------------------------- /examples/_pre_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/_pre_ms.py -------------------------------------------------------------------------------- /examples/_pre_nlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/_pre_nlp.py -------------------------------------------------------------------------------- /examples/ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/ae.py -------------------------------------------------------------------------------- /examples/cl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/cl.py -------------------------------------------------------------------------------- /examples/cl_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/cl_ddp.py -------------------------------------------------------------------------------- /examples/cl_ddp_spawn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/cl_ddp_spawn.py -------------------------------------------------------------------------------- /examples/cv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/cv.py -------------------------------------------------------------------------------- /examples/cv_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/cv_ddp.py -------------------------------------------------------------------------------- /examples/cv_ddp_spawn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/cv_ddp_spawn.py -------------------------------------------------------------------------------- /examples/dqn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/dqn.py -------------------------------------------------------------------------------- /examples/gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/gan.py -------------------------------------------------------------------------------- /examples/gnn_edge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/gnn_edge.py -------------------------------------------------------------------------------- /examples/gnn_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/gnn_graph.py -------------------------------------------------------------------------------- /examples/gnn_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/gnn_node.py -------------------------------------------------------------------------------- /examples/meta_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/meta_learning.py -------------------------------------------------------------------------------- /examples/nlp_baichuan_sft_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/nlp_baichuan_sft_lora.py -------------------------------------------------------------------------------- /examples/nlp_bert_mlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/nlp_bert_mlm.py -------------------------------------------------------------------------------- /examples/nlp_bert_seq_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/nlp_bert_seq_cls.py -------------------------------------------------------------------------------- /examples/nlp_chatglm2_sft_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/nlp_chatglm2_sft_lora.py -------------------------------------------------------------------------------- /examples/nlp_gpt_lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/nlp_gpt_lm.py -------------------------------------------------------------------------------- /examples/nlp_gpt_seq_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/nlp_gpt_seq_cls.py -------------------------------------------------------------------------------- /examples/nlp_gpt_zh_sft_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/nlp_gpt_zh_sft_adapter.py -------------------------------------------------------------------------------- /examples/nlp_gpt_zh_sft_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/nlp_gpt_zh_sft_lora.py -------------------------------------------------------------------------------- /examples/test_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/test_env.py -------------------------------------------------------------------------------- /examples/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/examples/vae.py -------------------------------------------------------------------------------- /mini_lightning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/mini_lightning/__init__.py -------------------------------------------------------------------------------- /mini_lightning/_mini_lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/mini_lightning/_mini_lightning.py -------------------------------------------------------------------------------- /mini_lightning/_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/mini_lightning/_types.py -------------------------------------------------------------------------------- /mini_lightning/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/mini_lightning/_utils.py -------------------------------------------------------------------------------- /mini_lightning/_visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/mini_lightning/_visualize.py -------------------------------------------------------------------------------- /mini_lightning/_warmup_lrs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/mini_lightning/_warmup_lrs.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_mini_lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/tests/test_mini_lightning.py -------------------------------------------------------------------------------- /tests/test_select_device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/tests/test_select_device.py -------------------------------------------------------------------------------- /tests/test_url.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/tests/test_url.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/tests/test_utils.py -------------------------------------------------------------------------------- /tests/test_visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/mini-lightning/HEAD/tests/test_visualize.py --------------------------------------------------------------------------------