Commit Graph

13 Commits (main)

Author SHA1 Message Date
fanqiNO1 eb4fa73b56
[Enhancement] Enhance inputs_to_half in DeepSpeedStrategy (#1400) 2023-11-02 17:19:42 +08:00
Zaida Zhou 8c934d2681
Refine error message (#1338) 2023-09-04 16:20:41 +08:00
Mashiro db32234241
[Feature] Add colossalai strategy (#1299) 2023-08-18 15:09:35 +08:00
Zaida Zhou 6b0d5a5f1d
[Docs] Add README for examples (#1295) 2023-08-07 22:53:57 +08:00
Desjajja 398d229910
Add a text translation example (#1283) 2023-08-07 15:33:48 +08:00
Mashiro 5c5ec8b168
Add a segmentation example (#1282) 2023-08-03 15:27:58 +08:00
Mashiro f4f2555324
Llama2 example (#1264) 2023-07-24 10:20:21 +08:00
Mashiro 399f76ffa8 [Experimental] Add support for FSDP (#1213) 2023-06-29 15:19:33 +08:00
Zaida Zhou ccd5dc8b18 [Experimental] Add FlexibleRunner and Strategies (#1183) 2023-06-29 15:19:33 +08:00
TankNee d4bb561ff8
Add a text classification in examples (#1122) 2023-05-09 15:48:47 +08:00
Mashiro a9b6753fbe
Make TTAModel compatible with FSDP (#611)
* Add build_runner_with_tta and PrepareTTAHook

* rename hook file

* support build tta runner with runner type

* add unit test

* Add build_runner_with_tta to index.rst

* minor refine

* Add runner test cast

* Fix unit test

* fix unit test

* tmp save

* pop None if key does not exist

* Fix is_model_wrapper and force register class in test_runner

* [Fix] Fix is_model_wrapper

* destroy group after ut

* register module in testcase

* pass through unit test

* fix as comment

* remove breakpoint

* remove mmengine/testing/runner_test_cast.py

* minor refine

* minor refine

* minor refine

* set default data preprocessor for model

* minor refine

* minor refine

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* fix lint

* Fix unit test

* replace  with  in ImgDataPreprocessor

* Fix as comment

* add inference tutorial in advanced tutorial

* update index.rst

* add tta example

* refine tta tutorial

* Add english tutorial

* add note for build_runner_with_tta

* Fix as comment

* add examples

* remove chinese comment

* Update docs/en/advanced_tutorials/test_time_augmentation.md

Co-authored-by: RangiLyu <lyuchqi@gmail.com>

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: RangiLyu <lyuchqi@gmail.com>
2022-12-27 21:39:43 +08:00
GPH 46209b8cbf
[Fix] Fix examples/distributed_training.py does not work in DDP (#700)
* Update distributed_training.py

Better example for DDP training

* Update distributed_training.py

* Update distributed_training.py

update according to reviwer's suggesstions.

* Update distributed_training.py

* Update distributed_training.py

The previous update copy data from main branch, its a mistake.
This update fix this mistake and the code is tested.
2022-11-09 22:04:08 +08:00
Zaida Zhou a5284165c6
Add the distributed training script (#487)
* Add the distributed training script

* fix md format
2022-08-30 19:05:51 +08:00