mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
mmpretrain/mmcls/models/heads
History
aso538 6b9e2b55dd
[Feature] Support LeViT backbone. (#1238) 
* 网络搭建完成、能正常推理

* 网络搭建完成、能正常推理

* 网络搭建完成、能正常推理

* 添加了模型转换未验证，配置文件 但有无法运行

* 模型转换、结构验证完成，可以推理出正确答案

* 推理精度与原论文一致 已完成转化

* 三个方法改为class 暂存

* 完成推理精度对齐 误差0.04

* 暂时使用的levit2mmcls

* 训练跑通，训练相关参数未对齐

* '训练相关参数对齐'参数'

* '修复训练时验证导致模型结构改变无法复原问题'

* '修复训练时验证导致模型结构改变无法复原问题'

* '添加mixup和labelsmooth'

* '配置文件补齐'

* 添加模型转换

* 添加meta文件

* 添加meta文件

* 删除demo.py测试文件

* 添加模型README文件

* docs文件回滚

* model-index删除末行空格

* 更新模型metafile

* 更新metafile

* 更新metafile

* 更新README和metafile

* 更新模型README

* 更新模型metafile

* Delete the model class and get_LeViT_model methods in the mmcls.models.backone.levit file

* Change the class name to Google Code Style

* use arch to provide default architectures

* use nn.Conv2d

* mmcv.cnn.fuse_conv_bn

* modify some details

* remove down_ops from the architectures.

* remove init_weight function

* Modify ambiguous variable names

* Change the drop_path in config to drop_path_rate

* Add unit test

* remove train function

* add unit test

* modify nn.norm1d to build_norm_layer

* update metafile and readme

* Update configs and LeViT implementations.

* Update README.

* Add docstring and update unit tests.

* Revert irrelative modification.

* Fix unit tests

* minor fix

Co-authored-by: mzr1996 <mzr1996@163.com>
 		2023-01-17 17:43:42 +08:00
..
__init__.py [Feature] Support LeViT backbone. (#1238) 2023-01-17 17:43:42 +08:00
base_head.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
cls_head.py [Feature] Support Multi-task. (#1229) 2022-12-30 10:36:00 +08:00
conformer_head.py [CI] Add Circle CI for mmcls 1.x. (#970) 2022-08-22 15:02:08 +08:00
deit_head.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
efficientformer_head.py [Feature] Add efficientformer Backbone for MMCls 1.x. (#1031) 2022-09-20 14:56:45 +08:00
levit_head.py [Feature] Support LeViT backbone. (#1238) 2023-01-17 17:43:42 +08:00
linear_head.py [Feature] Add efficientformer Backbone for MMCls 1.x. (#1031) 2022-09-20 14:56:45 +08:00
margin_head.py [Feature] Support getting model from the name defined in the model-index file. (#1236) 2022-12-06 17:00:22 +08:00
multi_label_cls_head.py [Feature] Support getting model from the name defined in the model-index file. (#1236) 2022-12-06 17:00:22 +08:00
multi_label_csra_head.py [Feature] Migrate CSRA head to 1.x. (#1177) 2022-11-21 10:39:16 +08:00
multi_label_linear_head.py Add multi_label heads 2022-07-18 11:11:13 +08:00
multi_task_head.py [Feature] Support Multi-task. (#1229) 2022-12-30 10:36:00 +08:00
stacked_head.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
vig_head.py [Feature] Support VIG Backbone. (#1304) 2023-01-17 16:55:56 +08:00
vision_transformer_head.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00