cuicheng01
|
4e6c36e269
|
Merge pull request #1833 from TingquanGao/dev/fix_dist_loss
fix calc metric error and calc loss error in distributed.
|
2022-04-22 22:41:57 +08:00 |
gaotingquan
|
b5a7eea504
|
fix: fix the bug that DistributedBatchSampler may sample repeatedly
|
2022-04-22 09:39:22 +00:00 |
|
gaotingquan
|
83ed5195c3
|
fix: set use_fp16_test to True when AMP O2 is enabled
|
2022-04-18 06:14:43 +00:00 |
|
gaotingquan
|
efde56ffc6
|
fix: only fp16 evaluation is supported when ampO2 is enabled
|
2022-04-13 12:14:14 +00:00 |
|
gaotingquan
|
474c918b27
|
fix: fix bug of batch_size statistics error
|
2022-04-13 09:19:30 +00:00 |
|
gaotingquan
|
c46189bad0
|
fix: fix bug about calc loss in dist
|
2022-04-12 06:56:44 +00:00 |
|
gaotingquan
|
b761325faa
|
fix: fp32 eval by default when enable amp
If you want to eval by fp16 when enable amp, please set Amp.use_fp16_test=True, False by default.
|
2022-04-02 19:22:10 +08:00 |
WangChen0902
|
7595ba6d70
|
add AFD (#1683)
* add AFD
|
2022-02-28 19:11:50 +08:00 |
|
gaotingquan
|
7040ce8314
|
refactor: change params to be consistent with amp
|
2022-01-25 11:58:07 +08:00 |
zhangbo9674
|
cd039a7b37
|
add save_dtype
|
2022-01-10 18:19:03 +08:00 |
|
zhangbo9674
|
d437bb0a7e
|
use fp32 to eval
|
2022-01-10 18:19:03 +08:00 |
|
zhangbo9674
|
bb19c1f7a6
|
fix eval bug
|
2022-01-10 18:19: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 |
|
gaotingquan
|
7732a69f1b
|
fix: fix key error in distillation
|
2021-12-16 18:21:08 +08:00 |
dongshuilong
|
f7ccc874e2
|
fix dali distributed eval bug
|
2021-11-16 11:09:21 +08:00 |
|
dongshuilong
|
278f6d8050
|
fig goooglenet distributed eval bug
|
2021-10-26 11:56:30 +00:00 |
|
dongshuilong
|
fd6f1ad2ca
|
fix clas distributed eval bug
|
2021-10-21 03:47:03 +00:00 |
|
dongshuilong
|
c93d638f4c
|
fix clas distributed eval bug
|
2021-10-20 11:22:37 +00:00 |
cuicheng01
|
fe6f614680
|
Update multilabel
|
2021-09-27 03:00:37 +00:00 |
|
cuicheng01
|
a90881c99f
|
Merge branch 'develop' into add_multilabel
|
2021-09-26 22:47:20 +08:00 |
|
cuicheng01
|
af9aae730e
|
add multilabel feature
|
2021-09-26 07:05:13 +00: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 |