22 Commits

Author SHA1 Message Date
HydrogenSulfate
61b4153907 add batch Tensor collate to simplify dali code in train/eval/retrival code 2022-10-13 12:05:50 +08:00
HydrogenSulfate
c03a66bfe4 Rename variable names that may be confused in retrieval.py 2022-08-26 06:16:11 +00:00
HydrogenSulfate
43410aa852 update config, add amp eval for retrieval 2022-06-21 23:03:56 +08:00
HydrogenSulfate
565053167d update code 2022-05-11 18:57:16 +08:00
HydrogenSulfate
88295413f5 add re-ranking code 2022-05-11 17:33:07 +08:00
HydrogenSulfate
16f910b451 add m1 and baseline config 2022-05-05 19:55:08 +08:00
HydrogenSulfate
0b1481402b merge devevelop 2022-05-05 14:42:06 +08:00
gaotingquan
fea9522a69 fix: dbg 2022-04-26 12:07:59 +08:00
gaotingquan
b5a7eea504
fix: fix the bug that DistributedBatchSampler may sample repeatedly 2022-04-22 09:39:22 +00:00
HydrogenSulfate
15a0afc67c update code 2022-04-22 15:52:16 +08:00
HydrogenSulfate
9dde9376fb update code 2022-04-22 15:25:11 +08:00
HydrogenSulfate
daf7eea23d fix code 2022-04-21 15:43:53 +08:00
HydrogenSulfate
41e1a86caf add center loss 2022-04-21 00:17:54 +08:00
HydrogenSulfate
af90cd7c59 update center loss config and related code 2022-04-12 13:07:53 +08:00
weishengyu
9de22673df dbg 2022-04-08 14:29:03 +08:00
littletomatodonkey
aea712cc87
add dist of rec model (#1574)
* add distillation loss func and rec distillation
2022-01-05 19:25:36 +08:00
weishengyu
af25e25640 modify format 2021-09-23 11:22:25 +08:00
dongshuilong
1a1eb3a1ae fix issues when gallery == query dataset 2021-08-27 06:25:27 +00:00
Bin Lu
d388d69a63
Update retrieval.py 2021-08-26 19:38:40 +08:00
Bin Lu
52663edf60
Update retrieval.py 2021-08-25 13:57:24 +08:00
dongshuilong
5bf74beb46 update according comments 2021-08-24 07:07:17 +00:00
dongshuilong
15f6f58139 refactor trainer v2 2021-08-24 03:02:55 +00:00