Commit Graph

21 Commits (741190a86485d60ff250aaf67ab7ab574ea004c0)

Author SHA1 Message Date
xiexinch 741190a864 support '--amp' option 2022-07-14 14:21:32 +08:00
linfangjian.vendor 3c54de06bb [Refactor] Refactor train test interface. 2022-06-09 03:01:43 +00:00
Rockey da6bb2c8c5 [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 0d978a8822 fix log repeat (#1267) 2022-02-08 13:40:19 +08:00
MengzhangLI 83100bb023 [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 2729a6d4d1 add multi-processes script (#1238) 2022-01-27 21:18:55 +08:00
Rockey 43ad37b478 [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 a7595c6ec6 [Fix] revise --option to --options #1140 2021-12-15 15:24:08 +08:00
Rockey 3a1f4285b9 [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 341ecc7633 [Enchance] Set a random seed when the user does not set a seed. (#1039) 2021-11-26 10:55:15 +08:00
Junjun2016 84edf6c190 fix load ckpt bug in swin (#928) 2021-09-30 07:50:44 -07:00
sennnnn 9ac053f49a [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 f3a49b56a4 [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 2fe0bddf5e [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 9849a8dc23 [Refactor]: Unified parameter initialization (#567)
* [Refactor]: Unified parameter initialization

* fixed pretrained
2021-06-16 21:41:29 -07:00
Yinhao Li 02b5d768aa [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 4423d32702 [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 8c0e093b31 Fixed slide inference (#90) 2020-08-25 20:01:01 +08:00
Jerry Jiarui XU 0f702f44af 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 fd100e02c4 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