Commit Graph

357 Commits (2495400a98a3da9bd8cfb3731a711f4c77cc56a3)

Author SHA1 Message Date
mzr1996 2495400a98 Merge branch 'dev' 2022-12-06 18:25:47 +08:00
Ma Zerun c737e65164
Bump version to v0.25.0. (#1244) 2022-12-06 18:18:13 +08:00
Ma Zerun 3e99395c29
[Fix] Fix a bug caused MMClsWandbHook stuck. (#1242) 2022-12-06 16:50:41 +08:00
TangYueran dc8691e889
[Feature] Support MLU backend. (#1159)
* Training on MLU is available
2022-11-15 17:02:16 +08:00
Hakjin Lee 05e4bc17b2
[Feature] Support Activation Checkpointing for ConvNeXt. (#1152)
* Support Activation Checkpointing for ConvNeXt

* Add test case

* Lint

* Add docstring
2022-11-14 17:08:31 +08:00
unseenme 4a3ad4f652
[Docs] Fixed typo in pytorch2torchscript.md (#1173)
* [Docs] Fixed typo in pytorch2torchscript.md

* [Docs] Fixed W293 blank line contains whitespace
2022-11-09 11:03:12 +08:00
JayChen aacaa7316c
[Docs] further detail for the doc for `ClassBalancedDataset`. (#901)
* futher detail for the doc for datasets/dataset_wrappers/ClassBalancedDataset

* fix
2022-11-02 17:52:58 +08:00
Ma Zerun 29c54dd9ac
Bump version to v0.24.1 (#1150) 2022-11-01 14:17:38 +08:00
wangjiangben-hw 17ed870fd1
[Feature] Support mmcls with NPU backend. (#1072)
* init npu

* Avoid to import latest MMCV code to be compatible with old verisons.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-10-24 11:45:14 +08:00
790475019 38040d5e05
[Fix] Fix performance issue in convnext DDP train. (#1098)
to fix performance issue in convnext DDP train
2022-10-17 10:10:19 +08:00
mzr1996 91b85bb4a5 Merge remote-tracking branch 'origin/dev' 2022-09-30 18:06:17 +08:00
Ma Zerun 7b45eb10cd
Bump version to v0.24.0 (#1067) 2022-09-30 18:03:53 +08:00
Hakjin Lee 1b4e9cd22a
[Improve] replace loop of progressbar in api/test. (#878) 2022-09-30 14:41:07 +08:00
takuoko 8c7b7b15a3
[Enhance] RepVGG for YOLOX-PAI. (#1025)
* repvgg add ppf for yoloxpai

* fix by review

* update stem_channels

* fix doc

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-09-30 14:20:53 +08:00
Fei Wang 0143e5fdb0
[Fix] val loader should not drop last by default. (#857) 2022-09-28 08:22:23 +08:00
takuoko 1047daa28e
[Feature] Support HorNet Backbone. (#1013)
* add hornet

* add hornet

* add hornet

* add hornet

* add hornet

* add hornet

* add hornet

* fix test for torch before 1.7.0

* del timm

* fix readme

* fix readme

* Update mmcls/models/backbones/hornet.py

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>

* fix docs

* fix docs

* s -> scale

* fix dims and dpr impl

* fix layer scale

* refactor gnconv

* add dw_cfg

* add convert tools

* update code

* update docs

* update readme

* update URLs

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-09-27 10:37:49 +08:00
takuoko 56589ee280
[Enhancement] Update VAN. (#1017)
* update van

* fix init

* b4 result

* update van

* keep old config

* keep old config

* fix metafile

* update VAN configs

* update example

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-09-27 09:44:40 +08:00
Hubert 6ebb3f77ad
[Fix] Fix attenstion clamp max params (#1034) 2022-09-26 14:12:51 +08:00
WRH 75ae8453ac
[Docs] Fix a typo in ImageClassifier (#1050) 2022-09-22 09:24:23 +08:00
Kai Hu 0b4a67dd31
[Refactor] Re-write get_sinusoid_encoding from third-party implementation. (#965) 2022-09-13 15:24:29 +08:00
Andrey Moskalenko 517bd3d34b
[Fix] Fix device mismatch in Swin-v2. (#976) 2022-09-01 18:03:49 +08:00
Jiahao Wang ec71d071d8
[Improve] Fixed typo in `RepVGG`. (#985)
* [Improve] Use `forward_dummy` to calculate FLOPS. (#953)

* fixed

Co-authored-by: Ming-Hsuan-Tu <alec.tu@acer.com>
2022-08-22 10:28:33 +08:00
mzr1996 5ad3bed2cd Merge remote-tracking branch 'origin/master' into dev 2022-08-22 10:12:24 +08:00
Ezra-Yu 6474ea2fc0
[Feature] Support EfficientFormer. (#954)
* add efficient backbone

* Update Readme and metafile

* Add unit tests

* fix confict

* fix lint

* update efficientformer head unit tests

* update README

* fix unit test

* fix Readme

* fix example

* fix typo

* recover api modification

* Update EfficiemtFormer Backbone

* fix unit tests

* add efficientformer to readme and model zoo
2022-08-16 23:38:08 +08:00
zzc98 7b16bcdd9b
[Feature] Support Stanford Cars dataset. (#893)
* feat: add stanford car dataset

* feat: add stanford car dataset

* feat: add stanford car dataset

* feat: add stanford car dataset

* feat: add stanford car dataset

* feat: add stanford car dataset

* Update links and using cars insteam of car

* place ependency scipy from runtime to optional

* Fix docstring

Co-authored-by: Ezra-Yu <1105212286@qq.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-16 11:14:17 +08:00
Ezra-Yu e54cfd6951
[Imporve] Using `train_step` instead of `forward` in PreciseBNHook (#964)
* fix precise BN hook when using MLU

* fix unit tests
2022-08-11 15:02:25 +08:00
Timothy Lim b366897889
[Docs] Refine the docstring of RegNet (#935)
* Update regnet.py

In the example comment to print out the different layers of outputs, we need to indicate the `out_indices` to (0,1,2,3) to see all backbone layers output as the default argument is (3,)

* Update regnet.py

following changes proposal of maintainer

* fix linting

* fix blank space for docs

* fix blank space for docs

* fix blank space for docs
2022-08-10 18:17:36 +08:00
Ming-Hsuan-Tu 90254a8455
[Improve] Use `forward_dummy` to calculate FLOPS. (#953) 2022-08-08 18:34:09 +08:00
JiayuXu 1a3d51acc2
[Feature] Support CSRA head. (#881)
* Support CSRA head.

* Add CSRA config.

* Improve training scheduler and Update cfg, ckpt, log

* Update metafile

* Rename config files and checkpoints

Co-authored-by: Ezra-Yu <1105212286@qq.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-04 18:15:51 +08:00
Ma Zerun b5bb86a357
[Fix] Fix the output position of Swin-Transformer. (#947)
* [Fix] Fix the output position of Swin-Transformer.

* Rename `downsample` argument to `do_downsample`.
2022-08-03 19:32:29 +08:00
Hubert 6ec38fe742
[Feature] Support Swin Transform V2. (#799)
* init rough try for modify

* Init swin transform v2

* lint

* reformat

* init config

* refactor

* update config

* fix test

* add doc

* refact

* add model meta

* rename config

* add doc

* fix meta model name

* restruct

* rename embed_bims to out_channels

* fix ut and update model
2022-08-03 17:33:08 +08:00
Ma Zerun 556fa567a8
[Feature] Support MViT and add checkpoints. (#924)
* [Feature] Support MViT.

* Add MViT configs and docs

* Add unit test

* Fix unit tests.
2022-08-02 15:20:16 +08:00
Ma Zerun 9300cc4e3f
Bump version to 0.23.2. (#937) 2022-07-28 14:15:23 +08:00
HinGwenWoong 00f0e0d0be
[Fix] Fix Albu crash bug. (#918)
* Fix albu BUG: using albu will cause the label from array(x) to array([x]) and crash the trainning

* Fix common

* Using copy incase potential bug in multi-label tasks

* Improve coding

* Improve code logic

* Add unit test

* Fix typo

* Fix yapf
2022-07-28 14:10:34 +08:00
Ma Zerun c03efeeea4
[Feature] Support MPS device. (#894)
* [Feature] Support MPS device.

* Add `auto_select_device`

* Add unit tests
2022-07-28 12:28:51 +08:00
Ma Zerun 313d35704f
Bump version to v0.23.1 (#863) 2022-06-02 21:22:54 +08:00
Ayush Thakur ccdbc82e39
[Feature] Dedicated MMClsWandbHook for MMClassification (Weights and Biases Integration) (#764)
* wandb integration

* visualize using wandb tables

* wandb tables enhanced

* Refactor MMClsWandbHook (#1)

* [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

* [Enhance] Improve CPE performance by reduce memory copy. (#762)

* [Feature] Support resize relative position embedding in `SwinTransformer`. (#749)

* [Feature]: Add resize rel pos embed

* [Refactor]: Create a separated resize_rel_pos_bias_table func

* [Refactor]: Refactor rel pos embed bias

* [Refactor]: Move interpolate into func

* Remove index buffer only when window_size changes

Co-authored-by: mzr1996 <mzr1996@163.com>

* [Feature] Add PoolFormer backbone and checkpoints. (#746)

* add PoolFormer

* fix some typos in PoolFormer

* fix lint error

* modify out_indices and gap

* fix typo

* fix lint

* fix typo

* fix typo in poolforemr README

* fix lint

* Update some paths

* Refactor freeze_stages method

* Add unit tests

* Fix lint

Co-authored-by: mzr1996 <mzr1996@163.com>

* Bump version to v0.22.1 (#785)

* [Docs] Refine API reference. (#774)

* [Docs] Refine API reference

* Add PoolFormer

* [Docs] Fix docs.

* [Enhance] Reduce the memory usage of unit tests for Swin-Transformer. (#759)

* [Feature] Support VAN. (#739)

* add van

* fix config

* add metafile

* add test

* model convert script

* fix review

* fix lint

* fix the configs and improve docs

* rm debug lines

* add VAN into api

Co-authored-by: Yu Zhaohui <1105212286@qq.com>

* [Feature] Support DenseNet. (#750)

* init add densenet implementation

* Add config and converted models

* update meta

* add test for memory efficient

* Add docs

* add doc for jit

* Update checkpoint path

* Update readthedocs

Co-authored-by: mzr1996 <mzr1996@163.com>

* [Fix] Use symbolic link in the API reference of Chinese docs.

* [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>

* [Fix] Fix lint and mmcv version requirement for IPU.

* Bump version to v0.23.0 (#809)

* Refacoter Wandb hook and refine docstring

Co-authored-by: XiaobingZhang <xiaobing.zhang@intel.com>
Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
Co-authored-by: Weihao Yu <1090924009@qq.com>
Co-authored-by: takuoko <to78314910@gmail.com>
Co-authored-by: Yu Zhaohui <1105212286@qq.com>
Co-authored-by: Hubert <42952108+yingfhu@users.noreply.github.com>
Co-authored-by: Hu Di <476658825@qq.com>
Co-authored-by: hudi <dihu@graphcore.ai>

* shuffle val data

* minor updates

* minor fix

Co-authored-by: Ma Zerun <mzr1996@163.com>
Co-authored-by: XiaobingZhang <xiaobing.zhang@intel.com>
Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
Co-authored-by: Weihao Yu <1090924009@qq.com>
Co-authored-by: takuoko <to78314910@gmail.com>
Co-authored-by: Yu Zhaohui <1105212286@qq.com>
Co-authored-by: Hubert <42952108+yingfhu@users.noreply.github.com>
Co-authored-by: Hu Di <476658825@qq.com>
Co-authored-by: hudi <dihu@graphcore.ai>
2022-06-02 17:58:49 +08:00
Ma Zerun 3e198e63e8
[Docs] Update install tutorials. (#854)
* [Docs] Update install tutorials.

* [Docs] Improve dataset docs

* Add option to show the results in demo.

* fix typo
2022-06-01 18:31:57 +08:00
Ma Zerun 7c5ddb1e5b
Bump version to v0.23.0 (#809) 2022-05-01 21:58:33 +08:00
mzr1996 1d6fbe0efe [Fix] Fix lint and mmcv version requirement for IPU. 2022-04-29 22:33:29 +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
Hubert 59292b315a
[Feature] Support DenseNet. (#750)
* init add densenet implementation

* Add config and converted models

* update meta

* add test for memory efficient

* Add docs

* add doc for jit

* Update checkpoint path

* Update readthedocs

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-28 13:06:03 +08:00
takuoko df6edd7f5a
[Feature] Support VAN. (#739)
* add van

* fix config

* add metafile

* add test

* model convert script

* fix review

* fix lint

* fix the configs and improve docs

* rm debug lines

* add VAN into api

Co-authored-by: Yu Zhaohui <1105212286@qq.com>
2022-04-28 09:35:17 +08:00
Ma Zerun 702c196514
[Docs] Refine API reference. (#774)
* [Docs] Refine API reference

* Add PoolFormer
2022-04-15 20:19:20 +08:00
mzr1996 29b882d4eb Merge branch 'dev' 2022-04-15 20:10:45 +08:00
Ma Zerun eda75d12ec
Bump version to v0.22.1 (#785) 2022-04-15 20:08:39 +08:00
Weihao Yu 2eb27d0d4e
[Feature] Add PoolFormer backbone and checkpoints. (#746)
* add PoolFormer

* fix some typos in PoolFormer

* fix lint error

* modify out_indices and gap

* fix typo

* fix lint

* fix typo

* fix typo in poolforemr README

* fix lint

* Update some paths

* Refactor freeze_stages method

* Add unit tests

* Fix lint

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-15 11:35:15 +08:00
Yuan Liu fbcf907dc6
[Feature] Support resize relative position embedding in `SwinTransformer`. (#749)
* [Feature]: Add resize rel pos embed

* [Refactor]: Create a separated resize_rel_pos_bias_table func

* [Refactor]: Refactor rel pos embed bias

* [Refactor]: Move interpolate into func

* Remove index buffer only when window_size changes

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-13 23:06:56 +08:00
XiaobingZhang 875195efe9
[Enhance] Improve CPE performance by reduce memory copy. (#762) 2022-04-02 11:12:20 +08:00
Ma Zerun 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