Commit Graph

615 Commits (3e99395c298c4fa144f6414f6d1821c49552d93d)
 

Author SHA1 Message Date
Ma Zerun 3e99395c29
[Fix] Fix a bug caused MMClsWandbHook stuck. (#1242) 2022-12-06 16:50:41 +08:00
Ma Zerun 96bb2dd219
[Fix] Fix the redundant `device_ids` in `tools/test.py`. (#1215) 2022-11-22 18:57:52 +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
mzr1996 8c63bb55a5 Merge branch 'dev' 2022-11-01 14:19:49 +08:00
Ma Zerun 29c54dd9ac
Bump version to v0.24.1 (#1150) 2022-11-01 14:17:38 +08:00
wangjiangben-hw dd664ffcd4
[Docs] Add NPU support page. (#1149)
* init readme

* [Docs] Finish the HUAWEI Ascend device support docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-01 14:10:18 +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
Ma Zerun a9489f6bd0
[GitHub] Update issue template and remove general question template. (#1087)
* [CI] Fix CI error from timm and PyTorch version. (#1076)

* [GitHub] Update issue template and remove general question template.

* Add branch check dropdown options.
2022-10-20 16:32:34 +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
Ma Zerun bcadb74d5b
[CI] Fix CI error from timm and PyTorch version. (#1076) 2022-10-10 11:46:49 +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
Mengyang Liu c5bcd4801a
[Docs] Fix typo in config.md. (#827) 2022-09-30 15:02:24 +08:00
Philipp Allgeuer 7dca27dd57
[Fix] Fix warning with `torch.meshgrid`. (#860)
* Fix warning with torch.meshgrid

* Add torch_meshgrid_ij wrapper

* Use `digit_version` instead of packaging package.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-30 15:01:36 +08:00
Hakjin Lee 1b4e9cd22a
[Improve] replace loop of progressbar in api/test. (#878) 2022-09-30 14:41:07 +08:00
HinGwenWoong 4eaaf89618
[Docs] Add version for torchvision to avoide error. (#903)
* Add version for torchvision

* Add version for torchvision
2022-09-30 14:31:45 +08:00
JongYoon Lim 2102d09dfc
[Docs] Fixed typo for `--out-dir` option of analyze_results.py. (#898) 2022-09-30 14:30:45 +08:00
tpoisonooo 27b0bd5a72
[Fix] Add matplotlib minimum version requriments. (#909) 2022-09-30 14:22:42 +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
Ezra-Yu 4d73607fb8
[Fix] Fix config.device bug in toturial. (#1059) 2022-09-28 08:17:26 +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
Songyang Zhang c94e9b3669
[Feature] Update the issue template with more links and emoji. (#1032)
* [Feature] update the issue template with more links and emoji

* fix lint error

* Use yaml format issue templates.

* Update template

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-26 14:05:26 +08:00
WRH 75ae8453ac
[Docs] Fix a typo in ImageClassifier (#1050) 2022-09-22 09:24:23 +08:00
Lei Lei a1b644bc75
[Doc] Fix typo in tutorial 2 (#1043) 2022-09-19 13:46:45 +08:00
Lei Lei 8d1bc557ab
[Docs] Fix typo for wrong reference. (#1036) 2022-09-16 14:24:47 +08:00
Kai Hu 0b4a67dd31
[Refactor] Re-write get_sinusoid_encoding from third-party implementation. (#965) 2022-09-13 15:24:29 +08:00
daquexian 6d8c91892c
[Improve] Upgrade onnxsim to v0.4.0. (#915) 2022-09-13 15:13:20 +08:00
mzr1996 982cab4138 Update README 2022-09-07 17:13:33 +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
mzr1996 71ef7bae85 Merge remote-tracking branch 'origin/dev' 2022-07-28 14:15:52 +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
mzr1996 11df205e39 [Fix] Remove duplicated wide-resnet metafile. 2022-07-05 11:35:44 +08:00
Ma Zerun 812f3d4536
[CI] Add test mim CI. (#879) 2022-06-22 17:28:02 +08:00