Commit Graph

403 Commits (24168fb25e11124e3c79d9796e0bc6d3ccd68871)

Author SHA1 Message Date
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 f83ff59c97 增加代码规范,删除一些无用的function 2023-01-04 13:42:00 +08:00
zh-hike 692b8d8c1f 增加data的代码复用及RecModel的修改,代码已跑通 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
kuizhiqing a49e11db98 only for npu 2022-12-07 14:14:22 +08:00
kuizhiqing 20848e6abc adapt roll op for swin transformer 2022-12-07 14:14:22 +08:00
zhangyubo0722 b61da414e5 add uniformer lst 2022-12-02 19:45:59 +08:00
cuicheng01 ec2029b821 Add NextViT code and docs 2022-11-23 14:59:55 +08:00
zhiqiu f04eb47fc3 replace strided_slice 2022-11-21 11:06:22 +08:00
gaotingquan cfab4c178a feat: add pplcnetv2_small and pplcnetv2_big 2022-11-18 13:14:37 +08:00
gaotingquan c1a8ce4c94 fix error that sync bn should not be used on cpu 2022-11-16 16:05:53 +08:00
cuicheng01 6d42c25cf2 fix_resnet_vd_nhwc_format 2022-11-16 10:40:40 +08:00
HydrogenSulfate 7d9f4dcb59 change Tensor.numpy()[0] to float(Tensor) for 0-D tensor case 2022-11-01 14:37:11 +08:00
gaotingquan a623851224 fix: defer update_res() calling until after the object's __init__ has completed execution 2022-10-26 16:19:06 +08:00
gaotingquan 316479d380 fix: add *args, **kwargs to init_net()
the argument may be "infer_add_softmax" used in exporting.
2022-10-26 16:19:06 +08:00
dongshuilong 944763d7a5 add fixmatch 2022-10-25 12:04:22 +08:00
HydrogenSulfate 3647da6d4a
Merge pull request #2383 from HydrogenSulfate/xbm_final
Add XBM model
2022-10-19 15:43:40 +08:00
gaotingquan a5607e7d09 fix: update pretrained url 2022-10-18 17:48:31 +08:00
gaotingquan defcb638c3 fix: update pretrained url 2022-10-18 17:48:31 +08:00
gaotingquan ad3c657e39 fix: update VAN_tiny to VAN_B0 2022-10-18 17:48:31 +08:00
HydrogenSulfate 2cfd8dd828 add xbm 2022-10-17 15:45:45 +08:00
gaotingquan b3ab418e80 refactor: mv the init_res of net to super().__init__()
Put the calling of update_res(), stop_after(), freeze_befor() to the construction method of the parent class. And the init_net() support to call stop_after() by config of Arch.
2022-09-28 20:13:19 +08:00
gaotingquan 7513c0b52f feat: support freeeze subnet
freeze subnet specified with the specified layer name by setting OutputTensor.stop_gradient=True
2022-09-28 20:13:19 +08:00
gaotingquan 0160b8dd0d fix 2022-09-27 11:56:42 +08:00
zengshao0622 9a01999860 fix cae_large 2022-09-13 08:05:07 +00:00
zengshao0622 90d591e410 update ppcls 2022-09-08 11:34:54 +00:00
zengshao0622 7b50ce6585 merge CAE 2022-09-08 08:11:25 +00:00