252 Commits

Author SHA1 Message Date
Zaida Zhou
447d3bba2c
Fix config of readthedocs (#1511) 2024-03-06 10:30:00 +08:00
Zaida Zhou
b5f2d5860d
Refine mmengine introduction (#1479) 2024-01-24 19:27:02 +08:00
Zaida Zhou
e4600a6993
[Docs] Add the usage of ProfilerHook (#1466) 2024-01-02 15:59:37 +08:00
XiwuChen
369f15e27a
[Docs] Fix nnodes in the doc of ddp training (#1462) 2024-01-02 10:42:58 +08:00
fanqiNO1
e461581e55
[Docs] Add build mmengine-lite from source (#1435) 2023-11-22 11:02:12 +08:00
fanqiNO1
6be0aeb777
[Feature] Support for installing mmengine without opencv (#1429) 2023-11-20 22:00:46 +08:00
Zaida Zhou
c65187c6b8
[Docs] Rename master to main (#1397) 2023-10-18 19:13:26 +08:00
fanqiNO1
bf30c444de
Update the version info (#1383) 2023-10-09 16:18:40 +08:00
Shu Liqiang
b8a31671a4
[Feature] Runner supports setting the number of iterations for per epoch (#1292) 2023-10-08 16:43:44 +08:00
Mashiro
95d875832a
[Enhance] Support for installing minimal runtime dependencies (#1362) 2023-10-08 16:38:40 +08:00
Zaida Zhou
9b94af42b8
[Docs] Update the usage of bitsandbytes in Chinese document (#1359) 2023-09-15 23:14:33 +08:00
Range King
9a19f8bc89
[Feature] Add AimVisBackend (#1347) 2023-09-09 11:39:48 +08:00
Range King
273fb2b333
[Feature] Add DVCLiveVisBackend (#1336) 2023-09-08 17:22:23 +08:00
Zaida Zhou
45ee96d0c4
[Docs] Add activation checkpointing usage (#1341) 2023-09-05 11:23:44 +08:00
Mashiro
19ab172b2d
Fix typos and documents of colossalai (#1315) 2023-08-22 16:13:55 +08:00
Mashiro
db32234241
[Feature] Add colossalai strategy (#1299) 2023-08-18 15:09:35 +08:00
Zaida Zhou
03ad86cfd2
[Docs] Add a image for neptune (#1312) 2023-08-18 10:48:55 +08:00
Theodore
43e308caaf
[Feature] Add NeptuneVisBackend (#1311) 2023-08-17 23:29:58 +08:00
Zaida Zhou
a483dba9d1
fix typo (#1298) 2023-08-08 10:49:13 +08:00
Zaida Zhou
a54e814bf8
[Docs] Fix unused parameters (#1288) 2023-08-03 15:45:30 +08:00
Mashiro
5c5ec8b168
Add a segmentation example (#1282) 2023-08-03 15:27:58 +08:00
Zaida Zhou
5ef75fd7a7
[Docs] Introduce how to customize distributed training settings (#1279) 2023-07-31 15:40:45 +08:00
KerwinKai
68360e7ce8
[Feature] Add parameter save_begin for CheckpointHook (#1271) 2023-07-25 19:21:21 +08:00
youkaichao
66d828d8d3
[Enhancement] Rename fast_conv_bn_eval to efficient_conv_bn_eval (#1251) 2023-07-15 22:13:17 +08:00
youkaichao
40e49ff747
[Feature] Enable fast conv bn eval (#1202) 2023-07-14 18:21:55 +08:00
Zaida Zhou
33e30b7cb6
[Docs] how to train a large model (#1228) 2023-07-05 18:20:07 +08:00
Mashiro
529bab815f
[Fix] Fix docs (#1233) 2023-07-04 22:20:30 +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
Mashiro
22aa46bf56
[Docs] Fix config doc (#1218) 2023-06-28 19:01:08 +08:00
Zaida Zhou
d03a1da9a9
[Docs] Add a document to introduce how to debug with vscode (#1212) 2023-06-27 16:46:47 +08:00
Mashiro
478c952a6d
[Refacor] Replace 'if base' with 'with read_base' context manager (#1207) 2023-06-25 13:53:19 +08:00
Mashiro
6ece63ed35
[Feature] Support Pure Python style Configuration File (#1071) 2023-06-16 12:52:07 +08:00
syo093c
0052873b41
[Docs] Fix typo in document (#1201) 2023-06-14 22:15:20 +08:00
Zaida Zhou
cf477d15a2
[Docs] Add the usage of clearml (#1180) 2023-06-01 21:54:30 +08:00
vugia truong
68414516aa
[Feature] Add vis backend for clearml (#878) (#1091) 2023-06-01 17:41:34 +08:00
Zaida Zhou
4a9e379c1a
[Feature] Support Sophia optimizers (#1170) 2023-05-30 15:44:06 +08:00
Zaida Zhou
691500dce6
[Docs] Move the usage of distributed training to a single document (#1171) 2023-05-28 20:10:16 +08:00
Xin Li
d59acfbd9f
[Docs] Translate data_element.md (#1067) 2023-05-23 12:59:29 +08:00
gy77
ec2e00ae90
[Docs] Fix a missing comma in tutorials/runner.md (#1146) 2023-05-16 14:20:44 +08:00
Zaida Zhou
70c28415db
[Docs] Move translation of infer.md to en (#1138) 2023-05-09 11:45:18 +08:00
XHr
3b7c70fa97
[Docs] Translate infer.md (#1121) 2023-05-09 10:53:51 +08:00
Mashiro
6ba667c8cf
[Fix] Save optimizer.state_dict() in cpu by default (#966) 2023-04-26 16:47:47 +08:00
Mashiro
1c01594c5c
[Docs] Update links (#1108) 2023-04-25 18:51:11 +08:00
Zaida Zhou
43165160e6
[Docs] Replace MMCls with MMPretrain in docs (#1096)
* [Docs] Replace MMCls with MMPretrain in docs

* fix format
2023-04-23 15:29:43 +08:00
luomaoling
5b9a1544b0
[Feature] Add torch_npu optimizer (#1079) 2023-04-21 15:15:10 +08:00
LEFTeyes
6b366f236c
[Docs] Translate tutorials/evaluation.md (#1053)
* [Docs] Translate tutorials/evaluation.md
2023-04-12 12:53:14 +08:00
Zaida Zhou
9207e84aa0
[Docs] Introduce the use of wandb and tensorboard (#912)
* [Docs] Introduce the use of wandb and tensorboard

* fix link

* Update docs/en/common_usage/visualize_training_log.md
2023-04-11 12:31:05 +08:00
sung-hwa kim
8bf1ecad38
[Feature] Add vis backend for MLflow. (#878)
* add vis mlflow backend
2023-04-07 16:35:41 +08:00
KerwinKai
5b35c5b6ad
[Feature] Publish models after training if published_keys is set in CheckpointHook (#987)
* add publish keys in checkpointhook and update hook.md file

* Update checkpoint_hook.py

To avoid `mypy` warning `mmengine/hooks/checkpoint_hook.py:358: error: Unsupported right operand type for in ("Optional[List[str]]") Found 1 error in 1 file (checked 224 source files)`

* Update hook.md

Try to avoid trim trailing whitespace waring in hook.md

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update checkpoint_hook.py

* Update docs/en/tutorials/hook.md

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

* Update mmengine/hooks/checkpoint_hook.py

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

* Update mmengine/hooks/checkpoint_hook.py

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

* Update mmengine/hooks/checkpoint_hook.py

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

* Update mmengine/hooks/checkpoint_hook.py

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

* Update hook.md

add 自动发布最好的和最后的权重

* Update mmengine/hooks/checkpoint_hook.py

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

* Update checkpoint_hook.py

add condition when the best checkpoints more than 1.

* Update mmengine/hooks/checkpoint_hook.py

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

* Update checkpoint_hook.py

delete re judge

* Update checkpoint_hook.py

* Update checkpoint_hook.py

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update checkpoint_hook.py

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update checkpoint_hook.py

* Update mmengine/hooks/checkpoint_hook.py

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

* Add Test for publish model

* Update checkpoint_hook.py

* Update test_checkpoint_hook.py

* Fix file to pass pre-commit check

* Update mmengine/hooks/checkpoint_hook.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Fix mypy warning

* rm not necessary line in checkpoint_hook.py

* Update mmengine/hooks/checkpoint_hook.py

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

* rm unnecessary messages add to message_hub

* Update mmengine/hooks/checkpoint_hook.py

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

* Update docs/zh_cn/tutorials/hook.md

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

* Update docs/zh_cn/tutorials/hook.md

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

* update checkpoint hook and hook.md file

* Apply suggestions from code review

* Apply suggestions from code review

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Update mmengine/hooks/checkpoint_hook.py

---------

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-29 10:25:14 +08:00