ccdafe32ca
update models graph of acc-latency
2023-01-17 14:57:04 +08:00
518291ecf0
fix
2023-01-12 19:31:49 +08:00
216291618c
add the pretrained url
2023-01-06 20:29:07 +08:00
f6df698c4f
micro_block -> layer_type
2023-01-06 20:29:07 +08:00
81de331e00
rename to re_parameterize() for re-parameterization nets
2023-01-06 20:29:07 +08:00
d4d3d01384
support re_parameterize
2023-01-06 20:29:07 +08:00
f82871b1f8
add copyright & reference
2023-01-06 20:29:07 +08:00
e41f2a8fee
support PCALighting
2023-01-06 20:29:07 +08:00
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
4292c1a90d
fix turtle import bug
2023-01-06 10:57:05 +08:00
b05c8b1b7e
fix(retrieval): remove max_iter in retrieval.py to avoid empty data
2023-01-05 18:44:22 +08:00
62c9d0f45d
remove train benchmark for dsnet
2023-01-05 16:05:08 +08:00
2f331d8588
use scipy.sparse to speed up local query expansion
2023-01-05 15:13:04 +08:00
6871d601ff
speed up local QE by sparse matmul
2023-01-05 15:13:04 +08:00
c6865e255e
refactor(retrieval): polish retrieval.py
2023-01-05 15:13:04 +08:00
97f99cd826
refactor(retrieval): polish retrieval.py
2023-01-05 15:13:04 +08:00
8542967bef
修改FixMatchCCSSL成CCSSL
2023-01-04 13:42:00 +08:00
65d42f825c
增加模型地址链接
2023-01-04 13:42:00 +08:00
e3f5fbc94a
修改CCSSL的README文件，CCSSL复现完成
2023-01-04 13:42:00 +08:00
f8978a2f0c
修改cifar100参数配置
2023-01-04 13:42:00 +08:00
d8f049ae76
增加代码规范，删除无用空格
2023-01-04 13:42:00 +08:00
c5a29dbaec
代码规范
2023-01-04 13:42:00 +08:00
f073e97d22
增加cifar100参数yaml
2023-01-04 13:42:00 +08:00
009f347d64
修改空格等代码规范
2023-01-04 13:42:00 +08:00
f83ff59c97
增加代码规范，删除一些无用的function
2023-01-04 13:42:00 +08:00
692b8d8c1f
增加data的代码复用及RecModel的修改，代码已跑通
2023-01-04 13:42:00 +08:00
4db132444d
增加RecModel配合WideResNet代码以及参数转换成RecModel
2023-01-04 13:42:00 +08:00
7823f340b1
cifar10_4000跑出论文精度代码
2023-01-04 13:42:00 +08:00
11f787f0fe
fix opencv requirement
2023-01-03 16:02:12 +08:00
65a7b57099
Update data_augmentation.md
2022-12-29 15:13:26 +08:00
d336944566
rm unset CUDA_VISIBLE_DEVICES before training
2022-12-29 11:44:27 +08:00
9d054b2a7e
support NHWC data format
2022-12-28 16:06:10 +08:00
169682002d
Some modifications
2022-12-28 16:01:54 +08:00
7badd5bf6c
add metrics and postprocess
2022-12-28 16:01:54 +08:00
a9cb41795a
modified DSNet docs
2022-12-22 14:55:46 +08:00
8d57bb31cf
modified DSNet config and TIPC txt
2022-12-22 14:55:46 +08:00
e069dedd7e
modified DSNet model names
2022-12-22 14:55:46 +08:00
28e094e097
add docs, config, model for DSNet
2022-12-22 14:55:46 +08:00
830852e745
modify configs
2022-12-22 14:54:25 +08:00
e238686e84
modify init
2022-12-22 14:54:25 +08:00
98d876a1b8
modify config
2022-12-22 14:54:25 +08:00
0140b38a80
add models
2022-12-22 14:54:25 +08:00
fdaf24ee78
add models
2022-12-22 14:54:25 +08:00
885e1bc05f
fix numpy int bug
2022-12-22 14:19:15 +08:00
7cdae10bcf
fix pksampler prob list shuffle bug
2022-12-22 12:01:05 +08:00
5ae888cf0d
rm the faiss version
2022-12-21 14:18:52 +08:00
97398d9f12
fix export pretrained_model when it is None
2022-12-16 18:46:21 +08:00
20725dd770
fix xbm.py
2022-12-16 18:46:21 +08:00
51d71016ac
support web link for 'Global.pretrained_model' when export models
2022-12-16 18:46:21 +08:00
0288285c91
feat(loss): add TripletAngularMarginLoss_XBM and refine XBM queue
2022-12-16 18:46:21 +08:00
gaotingquan
zh-hike
HydrogenSulfate
cuicheng01
zh-hike
zengshao0622
Wei Shengyu
zhangyubo0722
LittleMoon