4643 Commits

Author SHA1 Message Date
HydrogenSulfate
6871d601ff speed up local QE by sparse matmul 2023-01-05 15:13:04 +08:00
HydrogenSulfate
c6865e255e refactor(retrieval): polish retrieval.py 2023-01-05 15:13:04 +08:00
HydrogenSulfate
97f99cd826 refactor(retrieval): polish retrieval.py 2023-01-05 15:13:04 +08:00
zh-hike
8542967bef 修改FixMatchCCSSL成CCSSL 2023-01-04 13:42:00 +08:00
zh-hike
65d42f825c 增加模型地址链接 2023-01-04 13:42:00 +08:00
zh-hike
e3f5fbc94a 修改CCSSL的README文件,CCSSL复现完成 2023-01-04 13:42:00 +08:00
zh-hike
f8978a2f0c 修改cifar100参数配置 2023-01-04 13:42:00 +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
zengshao0622
11f787f0fe fix opencv requirement 2023-01-03 16:02:12 +08:00
Wei Shengyu
65a7b57099
Update data_augmentation.md 2022-12-29 15:13:26 +08:00
gaotingquan
d336944566 rm unset CUDA_VISIBLE_DEVICES before training 2022-12-29 11:44:27 +08:00
gaotingquan
9d054b2a7e support NHWC data format 2022-12-28 16:06:10 +08:00
zhangyubo0722
169682002d Some modifications 2022-12-28 16:01:54 +08:00
zhangyubo0722
7badd5bf6c add metrics and postprocess 2022-12-28 16:01:54 +08:00
LittleMoon
a9cb41795a modified DSNet docs 2022-12-22 14:55:46 +08:00
LittleMoon
8d57bb31cf modified DSNet config and TIPC txt 2022-12-22 14:55:46 +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
830852e745 modify configs 2022-12-22 14:54:25 +08:00
zhangyubo0722
e238686e84 modify init 2022-12-22 14:54:25 +08:00
zhangyubo0722
98d876a1b8 modify config 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
zengshao0622
885e1bc05f fix numpy int bug 2022-12-22 14:19:15 +08:00
zengshao0622
7cdae10bcf fix pksampler prob list shuffle bug 2022-12-22 12:01:05 +08:00
gaotingquan
5ae888cf0d rm the faiss version 2022-12-21 14:18:52 +08:00
HydrogenSulfate
97398d9f12 fix export pretrained_model when it is None 2022-12-16 18:46:21 +08:00
HydrogenSulfate
20725dd770 fix xbm.py 2022-12-16 18:46:21 +08:00
HydrogenSulfate
51d71016ac support web link for 'Global.pretrained_model' when export models 2022-12-16 18:46:21 +08:00
HydrogenSulfate
0288285c91 feat(loss): add TripletAngularMarginLoss_XBM and refine XBM queue 2022-12-16 18:46:21 +08:00
HydrogenSulfate
4060957669 fix(dali): add shuffle to prob_list in ExternalSource_RandomIdentity 2022-12-13 14:27:15 +08:00
HydrogenSulfate
b22752e2ea refactor DALI 2022-12-13 14:27:15 +08:00
kuizhiqing
49519f634e restore 2022-12-08 10:37:49 +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
gaotingquan
4f01e3bc4f add static training doc 2022-12-06 16:31:02 +08:00