658db80089
[Enhancement] Support deepspeed with flexible runner ( #1673 )
...
* [Feature] Support deepspeed with flexible runner
* [Fix] Reformat with yapf
* [Refacor] Rename configs
* [Fix] Reformat with yapf
* [Refactor] Remove unused keys
* [Refactor] Change the _base_ path
* [Refactor] Reformat
2023-06-29 10:16:27 +08:00
1ee9bbe050
[Docs] Update links ( #1457 )
...
* update links
* update readtherdocs
* update
* update
* fix lint
* update
* update
* update
* update cov branch
* update
* update
* update
2023-04-06 20:58:52 +08:00
b017670e1b
[Improve] Use PyTorch official `scaled_dot_product_attention` to accelerate `MultiheadAttention`. ( #1434 )
...
* [Improve] Use PyTorch official `scaled_dot_product_attention` to accelerate `MultiheadAttention`.
* Support `--local-rank` and `--amp` option for new version PyTorch.
* Fix imports and UT.
2023-03-29 15:50:44 +08:00
0979e78573
Rename the package name to `mmpretrain`.
2023-02-17 15:20:55 +08:00
7ec6062415
[Refactor] Unify the `--out` and `--dump` in `tools/test.py`. ( #1307 )
2023-02-09 14:05:03 +08:00
c73a5a8b15
[Fix] Fix bug in test tools. ( #1309 )
...
* fix append
* Update test.py
* Update test.py
2023-01-12 14:18:00 +08:00
97c4ae8805
[Improve] Update registries of mmcls. ( #1306 )
...
* [Improve] Update registries of mmcls.
* Update according to comments
2023-01-11 15:20:51 +08:00
9038c1c255
[Feature] Support TTA and add `--tta` in `tools/test.py`. ( #1161 )
...
* [Feature] Support TTA and add `--tta` in `tools/test.py`.
* Add unit tests.
* Rename the TTA model to `AverageClsScoreTTA`.
2022-12-30 11:46:17 +08:00
9999da646c
[Imporve] Update tools to enable `pin_memory` and `persistent_workers` by default. ( #1024 )
2022-09-14 11:57:32 +08:00
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
2b88df4484
[Refactor] Refactor the data flow. ( #989 )
...
* [Refactor] Refactor the data flow.
* Add comments about data preprocessor.
* Fix after mmengine folder structure refactoring.
2022-08-26 10:40:43 +08:00
24bcf069f8
[Refactor] Refactor dev scripts
2022-07-18 11:11:13 +08:00
5a3e736b4e
[Feature] Implement offline evaluation.
2022-07-18 11:11:13 +08:00
ee6e585e41
Mazerun/base classifier
2022-07-18 11:11:13 +08:00
522ab1fd84
[Feature] Add Visualization hook
2022-07-18 11:11:04 +08:00
efaa93aae1
[Refactor] Refactor entry scripts `train.py` and `test.py`.
2022-07-18 10:53:56 +08:00
877ea3080a
[Fix] Fix wrong `--local_rank`.
2022-05-17 21:10:25 +08:00
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
02c8f82147
[Enhance] Add extra dataloader settings in configs. ( #752 )
...
* Use `train_dataloader`, `val_dataloader` and `test_dataloader` settings
in the `data` field to specify different arguments.
* Fix bug
* Fix bug
2022-04-01 15:57:12 +08:00
a87ff7e153
[Enhance] Use `--a-b` instead of `--a_b` in arguments. ( #754 )
2022-03-30 19:16:28 +08:00
1a28f9ace6
[Feature] Add `evaluate` function for ConcatDataset. ( #650 )
...
* Add `evaluate` function for ConcatDataset
* Remove newline in log.
* Fix lint
* Specify mmcv version in Windows CI
2022-02-28 12:46:17 +08:00
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
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
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
f552419e45
[Enhance] Add setup multi-processing both in train and test. ( #671 )
2022-01-27 10:18:36 +08:00
e166ee7c0c
[Fix] Fix format error in `test.py` when metric returns `np.ndarray` ( #441 )
...
* Update test.py
Fix the output format error when metric-options is specified
* Update tools/test.py
Co-authored-by: Ma Zerun <mzr1996@163.com>
* Update test.py
delete "," after round(out, 2)
* Update test.py
delete else
* Update tools/test.py
Co-authored-by: Ma Zerun <mzr1996@163.com>
Co-authored-by: Ma Zerun <mzr1996@163.com>
2021-09-29 11:51:19 +08:00
cb09ed54e5
[Enhance] Support setting `--out-items` in `tools/test.py`. ( #437 )
...
* Support setting out_details in `tools/test.py`.
* Add assertion in `eval_metric` and `analyze_results`.
2021-09-15 11:06:16 +08:00
ae40c45c55
[Enhance] Rename the option `--options` in some tools to `--cfg-options`. ( #425 )
2021-09-01 12:49:03 +08:00
f9eb9b409b
[Docs] Add Copyright information. ( #413 )
2021-08-17 19:52:42 +08:00
aad796ae6f
Modify test tools and add some new tools ( #322 )
...
* Refactor tools folder structure.
* Modify tools/test.py and add eval_metric.py to analysis test output.
* Add new tools `analyze_logs.py` and `print_config.py`.
* Add comment for analysis_tools functions.
2021-06-30 20:51:16 +08:00
b99bd4fa88
Fix bug for CPU training ( #286 )
...
* remove MMDataParallel when using cpu
* support cpu testing
* fix lint
2021-06-12 22:26:33 +08:00
6de635a81c
[Bug] Missing test data when num_imgs can not be evenly divided by num_gpus ( #299 )
...
* fix bug in test
* remove unneccesary code
2021-06-11 15:53:23 +08:00
3affc481c8
[Fix] check for CLASSES in checkpoint meta ( #207 )
...
- check for CLASSES in checkpoint meta when key meta does not exists
2021-04-15 22:19:23 +08:00
e76c5a368d
[Feature] Support fp16 training ( #178 )
...
* change mmcls fp16 to mmcv hook
* support fp16
* clean unnessary stuff
2021-03-17 15:53:55 +08:00
fb11a23cfe
Show results in test api ( #162 )
...
* Initial support show in single_gpu_test
* Add result_show
* Add ori_filename meta_key
* test ori_filename
* Add show, show-dir, show-options
2021-02-25 16:00:22 +08:00
ddc2a14177
[Feature] Add --eval-options in test.py ( #158 )
...
* add --eval-options in test.py
* fix typo
* revise according to commnets
2021-02-05 17:46:43 +08:00
a7d4739d2b
[Fix] Fix test.py ( #155 )
...
* modeify test.py
* modeify test.py
* revise according to comments
2021-01-31 15:53:43 +08:00
e75f2b7c35
add argument to override configs ( #91 )
2020-11-20 13:50:28 +08:00
9547e7b7a5
Add model inference ( #16 )
...
* add model inference on single image
* rm --eval
* revise doc
* add inference tool and demo
* fix linting
* rename inference_image to inference_model
* infer pred_label and pred_score
* fix linting
* add docstr for inference
* add remove_keys
* add doc for inference
* dump results rather than outputs
* add class_names
* add related infer scripts
* add demo image and the first part of colab tutorial
* conduct evaluation in dataset
* return lst in simple_test
* compuate topk accuracy with numpy
* return outputs in test api
* merge inference and evaluation tool
* fix typo
* rm gt_labels in test conifg
* get gt_labels during evaluation
* sperate the ipython notebook to another PR
* return tensor for onnx_export
* detach var in simple_test
* rm inference script
* rm inference script
* construct data dict to replace LoadImage
* print first predicted result if args.out is None
* modify test_pipeline in inference
* refactor class_names of imagenet
* set class_to_idx as a property in base dataset
* output pred_class during inference
* remove unused docstr
2020-09-30 19:00:20 +08:00
8d3acce307
Add ResNeSt ( #25 )
...
* Add ResNeSt
* fixed test
* refactor
* add ResNeSt base
* update modelzoo
* update modelzoo
* Add S-200,S269 _base_
2020-09-22 10:41:51 +08:00
fbc68bf266
Add classifiers, heads, necks and losses
2020-07-07 19:32:06 +08:00
a0d4f95725
Add missing mmcls.core
2020-05-27 11:37:16 +08:00
a99e6eb10d
init commit
2020-05-21 21:21:43 +08:00
fanqiNO1
Yixiao Fang
Ma Zerun
CSH
Ezra-Yu
Hu Di
Shane Zhao
WRH
LXXXXR
agim-a
David de la Iglesia Castro
Lei Yang
Jerry Jiarui XU
yanglei
chenkai