gaotingquan
|
2a61ac65bf
|
debug
|
2023-03-10 16:56:55 +08:00 |
|
gaotingquan
|
9beb154bc3
|
support ShiTu
|
2023-03-10 16:56:55 +08:00 |
zengshao0622
|
1c68b63a66
|
fix some nots
|
2023-02-15 12:00:55 +08:00 |
HydrogenSulfate
|
b05c8b1b7e
|
fix(retrieval): remove max_iter in retrieval.py to avoid empty data
|
2023-01-05 18:44:22 +08:00 |
|
HydrogenSulfate
|
2f331d8588
|
use scipy.sparse to speed up local query expansion
|
2023-01-05 15:13:04 +08:00 |
|
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 |
|
HydrogenSulfate
|
7d9f4dcb59
|
change Tensor.numpy()[0] to float(Tensor) for 0-D tensor case
|
2022-11-01 14:37:11 +08:00 |
|
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 |