355 Commits

Author SHA1 Message Date
zh-hike
e8bf35d1b4 fix foundationvit load fail 2023-02-07 10:46:25 +08:00
zhangyubo0722
83fc10cd0f modify uniformer link 2023-02-06 19:28:30 +08:00
zhangyubo0722
a170400d20 modify init 2023-02-01 10:43:11 +08:00
zhangyubo0722
79cbd7350a Aesthetic 2023-02-01 10:43:11 +08:00
HydrogenSulfate
4fdcda7c60 fix bug in randaug, train_progressive and efficientnet_v2 2023-01-31 21:15:22 +08:00
HydrogenSulfate
7e12c73e0a polish progressive training code 2023-01-31 21:15:22 +08:00
HydrogenSulfate
8b8e0431ed fix AttrDict import path 2023-01-31 21:15:22 +08:00
HydrogenSulfate
3a8b56801d feat(model): add EfficientNetV2 code and fix AttrDict BUG 2023-01-31 21:15:22 +08:00
zhangyubo0722
b889101fd1 add docs and config 2023-01-31 19:57:43 +08:00
zhangyubo0722
373a026a83 modify efficientnet_variant 2023-01-31 19:57:43 +08:00
zhangyubo0722
54f7dd7484 watermark 2023-01-31 19:57:43 +08:00
zhangyubo0722
add3ea238c modify tipc 2023-01-20 14:08:06 +08:00
zhangyubo0722
9ba23527b4 add swinv2 2023-01-20 14:08:06 +08:00
zhangyubo0722
4786b2c96b add swinv2 2023-01-20 14:08:06 +08:00
zhangyubo0722
74de24f621 add swinv2 2023-01-20 14:08:06 +08:00
zhangyubo0722
a5a3b0f567 Delete unnecessary files 2023-01-20 14:08:06 +08:00
zhangyubo0722
272e75365b modify swinv2 2023-01-20 14:08:06 +08:00
zhangyubo0722
65e7a41a1d add docs 2023-01-20 14:08:06 +08:00
zhangyubo0722
7e616a16b5 modify init 2023-01-20 14:08:06 +08:00
zhangyubo0722
f5e8df0e63 Some modifications 2023-01-20 14:08:06 +08:00
zhangyubo0722
da95893bf2 supplement init 2023-01-20 14:08:06 +08:00
zhangyubo0722
6994050983 add SwinTransformerV2 2023-01-20 14:08:06 +08:00
zhangyubo0722
0a56120855 add swinV1 22k weights 2023-01-20 14:07:04 +08:00
zhangyubo0722
3672d1f227 add swinV1 22k weights 2023-01-20 14:07:04 +08:00
zengshao0622
5544dbaf8a add ViT model for foundation models forward 2023-01-19 17:42:45 +08:00
gaotingquan
216291618c add the pretrained url 2023-01-06 20:29:07 +08:00
gaotingquan
f6df698c4f micro_block -> layer_type 2023-01-06 20:29:07 +08:00
gaotingquan
81de331e00 rename to re_parameterize() for re-parameterization nets 2023-01-06 20:29:07 +08:00
gaotingquan
d4d3d01384 support re_parameterize 2023-01-06 20:29:07 +08:00
gaotingquan
f82871b1f8 add copyright & reference 2023-01-06 20:29:07 +08:00
gaotingquan
963dcb9cac support dbb module for ResNet
1. add DiverseBranchBlock module;
2. ResNet support dbb version using DiverseBranchBlock by setting micro_block="DiverseBranchBlock";
3. ResNet support official vb version by setting use_first_short_conv=False;
4. add ResNet18_dbb training config ResNet18_dbb.yaml.
2023-01-06 20:29:07 +08:00
zh-hike
d8f049ae76 增加代码规范,删除无用空格 2023-01-04 13:42:00 +08:00
zh-hike
c5a29dbaec 代码规范 2023-01-04 13:42:00 +08:00
zh-hike
f073e97d22 增加cifar100参数yaml 2023-01-04 13:42:00 +08:00
zh-hike
009f347d64 修改空格等代码规范 2023-01-04 13:42:00 +08:00
zh-hike
4db132444d 增加RecModel配合WideResNet代码以及参数转换成RecModel 2023-01-04 13:42:00 +08:00
zh-hike
7823f340b1 cifar10_4000跑出论文精度代码 2023-01-04 13:42:00 +08:00
gaotingquan
9d054b2a7e support NHWC data format 2022-12-28 16:06:10 +08:00
LittleMoon
e069dedd7e modified DSNet model names 2022-12-22 14:55:46 +08:00
LittleMoon
28e094e097 add docs, config, model for DSNet 2022-12-22 14:55:46 +08:00
zhangyubo0722
e238686e84 modify init 2022-12-22 14:54:25 +08:00
zhangyubo0722
0140b38a80 add models 2022-12-22 14:54:25 +08:00
zhangyubo0722
fdaf24ee78 add models 2022-12-22 14:54:25 +08:00
kuizhiqing
17a4b50ae2 rm singleton 2022-12-08 10:37:49 +08:00
kuizhiqing
9d69b63c6d fix ct singleton 2022-12-07 14:14:22 +08:00
kuizhiqing
7cbc695227 use np for cpu tensor 2022-12-07 14:14:22 +08:00
kuizhiqing
19ed923799 fix typo 2022-12-07 14:14:22 +08:00
kuizhiqing
d70668cb6f fix typo 2022-12-07 14:14:22 +08:00
kuizhiqing
a37cee70bf remove global vars 2022-12-07 14:14:22 +08:00
kuizhiqing
fd4a97d144 lazy roll 2022-12-07 14:14:22 +08:00