Aston.He
7628a61f92
feat(mlu): Support PyTorch backend on MLU. ( #1515 )
...
* feat(mlu): Support PyTorch backend on MLU.
* fix redundant device variable.
* Update mmseg/apis/train.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update comments.
* Update mmseg/apis/train.py
* Update is_mlu_available flag.
* align util_distribution.py to mmdet.
* align util_distribution.py to mmdet.
* add build_dp, build_ddp testcase.
* Update mmseg/utils/util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmseg/utils/util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmseg/utils/util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* add mmcv version check for mlu device.
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-25 18:11:42 +08:00
Rockey
7dba7a2620
[Feature]: Add diff seeds to diff ranks and set torch seed in worker_init_fn ( #1362 )
2022-03-10 09:41:16 +08:00
MengzhangLI
af88b868a6
fix log repeat ( #1267 )
2022-02-08 13:40:19 +08:00
MengzhangLI
622f28eea3
[Fix] Fix bug in non-distributed multi-gpu training/testing ( #1247 )
...
* Fix bug in non-distributed training
* Fix bug in non-distributed testing
* delete uncomment lines
* add args.gpus
2022-01-28 10:38:16 +08:00
MengzhangLI
8ea3d64c65
add multi-processes script ( #1238 )
2022-01-27 21:18:55 +08:00
Rockey
78fdf65382
[Feature] add auto resume ( #1172 )
...
* [Feature] add auto resume
* Update mmseg/utils/find_latest_checkpoint.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/utils/find_latest_checkpoint.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* modify docstring
* Update mmseg/utils/find_latest_checkpoint.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* add copyright
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
2022-01-11 12:27:24 +08:00
Rockey
4e9a95d4f2
[Fix] revise --option to --options #1140
2021-12-15 15:24:08 +08:00
Rockey
6f844429ea
[Feature] Change options to cfg-option ( #1129 )
...
* [Feature] Change option to cfg-option
* add expire date and fix the docs
* modify docstring
2021-12-14 19:11:52 +08:00
Junjun2016
f0426c97d7
[Enchance] Set a random seed when the user does not set a seed. ( #1039 )
2021-11-26 10:55:15 +08:00
Junjun2016
10886b00f0
fix load ckpt bug in swin ( #928 )
2021-09-30 07:50:44 -07:00
sennnnn
cae715a4b6
[Fix] Convert SyncBN to BN when training on DP ( #772 )
...
* [Fix] Convert SyncBN to BN when training on DP.
* Modify SyncBN2BN.
* Add SyncBN2BN unit test.
* Resolve some comments.
* use mmcv official revert_sync_batchnorm
* Remove local syncbn2bn unit tests.
* Update mmcv version.
* Fix bugs of gather model tools.
* Modify warnings.
* Modify docker mmcv version.
* Update mmcv version table.
2021-09-15 09:39:37 -07:00
FreyWang
c4c2fdcff3
[Fix] Fix iter bug when resuming checkpoint in distributed train ( #866 )
...
* [Fix] Fix iter bug when resuming checkpoint in distributed train
* fix lint error
Signed-off-by: FreyWang <wangwxyz@qq.com>
2021-09-11 14:33:01 +08:00
Junjun2016
441be4e435
[Dcos] Add header for files ( #796 )
...
* Add header for files
* Delete header in config files
2021-08-16 23:16:55 -07:00
Jerry Jiarui XU
0c5b026db1
[Refactor]: Unified parameter initialization ( #567 )
...
* [Refactor]: Unified parameter initialization
* fixed pretrained
2021-06-16 21:41:29 -07:00
Yinhao Li
9dd3e15f83
[feature]: Able to use save_best option ( #575 )
...
* Add save_best option in eval_hook.
* Update meta to fix best model can not test bug
* refactor with _do_evaluate
* remove redundent
* add meta
Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2021-06-02 16:25:26 -07:00
Jerry Jiarui XU
faaf29e668
[Improvement] Move train_cfg/test_cfg inside model ( #341 )
...
* [Improvement] Move train_cfg/test_cfg inside model
* fixed config dict
* fixed config dict
2021-01-19 17:06:23 -08:00
Jerry Jiarui XU
bafc0e5db6
Fixed slide inference ( #90 )
2020-08-25 20:01:01 +08:00
Jerry Jiarui XU
4e29452dbe
Auto get version info and git hash ( #55 )
...
* Auto get version info and git hash
* bump 0.5.1 and update doc
* fixed docs
* Add change log
2020-08-11 19:23:35 +08:00
Evgeny Nizhibitsky
00f56eb00f
Add load-from flag ( #33 )
...
* Add load-from flag
* minor update
Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-08-06 21:17:49 +08:00
Jiarui XU
b2724da80b
init commit
2020-07-10 02:39:01 +08:00