Commit Graph

1455 Commits (692b8d8c1f3977973f7c8f00006367d377bb0e52)

Author SHA1 Message Date
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
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 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
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 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
zhangyubo0722 b61da414e5 add uniformer lst 2022-12-02 19:45:59 +08:00
HydrogenSulfate d7f73dd8ed add relabel=True in ResNet50_ReID to avoid label excess 2022-12-02 15:00:46 +08:00
cuicheng01 5b0a47bcdf fix NextViT config 2022-11-30 16:25:30 +08:00
HydrogenSulfate 6cc6540ca5 add different seed for workers and replicas 2022-11-24 11:51:13 +08:00
HydrogenSulfate b542416d67 add random shuffle in PKSampler 2022-11-24 11:51:13 +08:00
cuicheng01 ec2029b821 Add NextViT code and docs 2022-11-23 14:59:55 +08:00
HydrogenSulfate 1ad4167fdc fix engine.py 2022-11-22 16:47:47 +08:00
HydrogenSulfate 25ed25372a fix engine.py 2022-11-22 16:47:47 +08:00
HydrogenSulfate c07758b331 fix engine.py 2022-11-22 16:47:47 +08:00
HydrogenSulfate fa52acd614 remove redundant import in pksampler 2022-11-22 16:47:47 +08:00
HydrogenSulfate c26adddf33 remove redundant code in pksampler 2022-11-22 16:47:47 +08:00
HydrogenSulfate 7ff257ea69 fix random seed bug for pksampler in DDP 2022-11-22 16:47:47 +08:00
zhiqiu f04eb47fc3 replace strided_slice 2022-11-21 11:06:22 +08:00
user3984 aed02a213e update pefd 2022-11-18 14:04:07 +08:00
user3984 7ee8471dfe add pefd 2022-11-18 14:04:07 +08:00
weishengyu faca0ba75d add fixmatch config using 4gpus 2022-11-18 13:46:43 +08:00
gaotingquan cfab4c178a feat: add pplcnetv2_small and pplcnetv2_big 2022-11-18 13:14:37 +08:00
cuicheng01 1a8590a9bd fix TprAtFpr metrics 2022-11-18 13:11:29 +08:00