16 Commits

Author SHA1 Message Date
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
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