Ma Zerun
97c4ae8805
[Improve] Update registries of mmcls. ( #1306 )
...
* [Improve] Update registries of mmcls.
* Update according to comments
2023-01-11 15:20:51 +08:00
Ma Zerun
29c46c8af2
[Docs] Add runtime configuration docs. ( #1128 )
...
* [Docs] Add runtime configuration docs.
* Fix grammar errors.
* Imporve docs according to comments
2022-11-02 10:59:59 +08:00
Ma Zerun
9999da646c
[Imporve] Update tools to enable `pin_memory` and `persistent_workers` by default. ( #1024 )
2022-09-14 11:57:32 +08:00
Ezra-Yu
0e3e8e91cc
[Refactor] Refactor Config Doc. ( #987 )
...
* refactor config doc
* refine
* Fix the config docs.
* update CN config
* Fix some grammar erros.
* Fix Chinese docs.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-29 11:10:05 +08:00
Ma Zerun
7f0502867b
[Improve] Add `--amp`, `--resume` and `--auto-scale-lr` options in `tools/train.py`. ( #914 )
2022-07-21 17:34:04 +08:00
mzr1996
ee6e585e41
Mazerun/base classifier
2022-07-18 11:11:13 +08:00
mzr1996
efaa93aae1
[Refactor] Refactor entry scripts `train.py` and `test.py`.
2022-07-18 10:53:56 +08:00
mzr1996
877ea3080a
[Fix] Fix wrong `--local_rank`.
2022-05-17 21:10:25 +08:00
Hu Di
b4eefe4794
[Enhance] Support training on IPU and add fine-tuning configs of ViT. ( #723 )
...
* implement training and evaluation on IPU
* fp16 SOTA
* Tput reaches 5600
* 123
* add poptorch dataloder
* change ipu_replicas to ipu-replicas
* add noqa to config long line(website)
* remove ipu dataloder test code
* del one blank line in test_builder
* refine the dataloder initialization
* fix a typo
* refine args for dataloder
* remove an annoted line
* process one more conflict
* adjust code structure in mmcv.ipu
* adjust ipu code structure in mmcv
* IPUDataloader to IPUDataLoader
* align with mmcv
* adjust according to mmcv
* mmcv code structre fixed
Co-authored-by: hudi <dihu@graphcore.ai>
2022-04-29 22:22:19 +08:00
Ma Zerun
a87ff7e153
[Enhance] Use `--a-b` instead of `--a_b` in arguments. ( #754 )
2022-03-30 19:16:28 +08:00
Hubert
995f2912ab
[Feature] Add diff seeds to diff ranks. ( #744 )
...
* [Feature] Add diff seeds to diff ranks.
* lint
2022-03-25 15:31:13 +08:00
Ezra-Yu
7fcaedcbfb
[Feature] Add ResNetV1c. ( #692 )
...
* add ResNetV1c
* add unit tests
* fix lint
* update docstring
* fix lint
2022-02-23 11:36:33 +08:00
Ma Zerun
44e9902979
[Refactor] Use `--gpu-id` instead of `--gpu-ids` in non-distributed multi-gpu training/testing. ( #688 )
2022-02-17 02:17:01 +08:00
Ma Zerun
bca695b684
[Refactor] Remove deprecation. ( #633 )
...
* Remove deprecated `--options` in some tools
* Remove deprecated eval hooks and fp16 hooks
* Remove deprecated mixup&cutmix, pretrained and return_tuple arguments.
* Remove deprecated `deprecated_options` in multi label dataset
* Remove deprecated `formating.py`
* Remove deprecated default mode of `LabelSmoothLoss`
* Revert pretrained argument, and remove it's deprecation sign.
* Update unit test
2022-02-10 09:30:18 +08:00
Ma Zerun
8ef5aeda40
[Enhance] New-style CPU training and inference. ( #674 )
...
* [Enhance] New-style CPU training and inference.
* Add version check in CPU training/test
2022-01-30 20:49:54 +08:00
Ma Zerun
f552419e45
[Enhance] Add setup multi-processing both in train and test. ( #671 )
2022-01-27 10:18:36 +08:00
imyhxy
32d606e1a1
[Refactor] Save class names in best checkpoint created by evaluation hook ( #641 )
...
* Save class names in best checkpoint created by evaluation hook
* Save meta info into runner instaed of checkpoint_config
2022-01-13 12:20:59 +08:00
Ma Zerun
d25a78d547
[Enchance] Set a random seed when the user does not set a seed. ( #554 )
2021-12-02 18:09:55 +08:00
Ma Zerun
ae40c45c55
[Enhance] Rename the option `--options` in some tools to `--cfg-options`. ( #425 )
2021-09-01 12:49:03 +08:00
Ma Zerun
f9eb9b409b
[Docs] Add Copyright information. ( #413 )
2021-08-17 19:52:42 +08:00
Miao Zheng
4ca21c7d03
[WIP] Refactoring weights initialization ( #270 )
...
* [WIP] Refactoring weights initialization
* fix lint and constant init cfg
* fix pretrained bug
* fix typo
* fix isort
* revise model utils
2021-06-10 10:54:34 +08:00
LXXXXR
5066e32306
dump config before training ( #282 )
2021-06-04 16:26:13 +08:00
WRH
9be435846c
Support training on CPU ( #219 )
...
* draft
* add parameter for training tools
* Update .pre-commit-config.yaml
2021-04-26 13:56:45 +08:00
LXXXXR
f355f15485
Fix bug in gpu_ids in distributed training ( #107 )
...
* update gpu_ids in distributed training
* move linear scaling rule after getting correct gpu_ids
* Remove support for autoscale_lr
2020-12-09 16:28:03 +08:00
yanglei
fbc68bf266
Add classifiers, heads, necks and losses
2020-07-07 19:32:06 +08:00
chenkai
a99e6eb10d
init commit
2020-05-21 21:21:43 +08:00