Commit Graph

35 Commits (18b25e3e868d79e6cd2b7a3c10cd897a62ceaf6e)

Author SHA1 Message Date
gaotingquan 7040ce8314 refactor: change params to be consistent with amp 2022-01-25 11:58:07 +08:00
Tingquan Gao ce166b6ced docs: fix invalid links 2022-01-17 21:14:38 +08:00
Wei Shengyu 0f35f706b6
Fix static training speed (#1590)
* fix training speed

* update config setting method
2021-12-23 11:13:51 +08:00
gaotingquan 720c451bc4 fix: fix batch size to 64 * 4 2021-11-10 00:16:23 +08:00
gaotingquan ba2dd01a13 refactor: deprecate MixCELoss 2021-10-27 19:47:43 +08:00
Walter f82d109524
Merge pull request #1053 from zhiqiu/dev/refine_dygraph_amp_config
refine dygraph amp config
2021-07-23 10:31:27 +08:00
littletomatodonkey fa9221479c fix fp16 inference 2021-07-22 13:25:53 +08:00
zhiqiu bb1376db45 add flags setting 2021-07-16 13:26:03 +00:00
littletomatodonkey a9f35981e9
fix fp16 config for dyg (#1052) 2021-07-16 18:08:16 +08:00
littletomatodonkey c2054df0e9
add r50 fp16 dyg (#1047) 2021-07-15 11:16:30 +08:00
littletomatodonkey 9d9cd3719e
add static training (#1037)
* add static training

* fix typo

* add se fp16

* rm note

* fix loader

* fix cfg
2021-07-15 10:30:07 +08:00
cuicheng01 592f5b10f5 support batch mix 2021-07-13 06:38:01 +00:00
littletomatodonkey dade4ef46b
fix class num (#989) 2021-06-29 20:28:55 +08:00
littletomatodonkey 558c2dd452 fix yaml 2021-06-29 10:42:39 +00:00
Aurelius84 d5c1700faf support to_static for benchmark 2021-06-21 06:14:06 +00:00
littletomatodonkey 215872506a
fix yaml note (#872) 2021-06-16 23:53:44 +08:00
cuicheng01 03b0ceef0f Update model_zoo train ImageNet configs 2021-06-11 07:59:58 +00:00
littletomatodonkey 63cf3ba4d1
fix imread (#824)
* fix imread

* rm unused decode

* rm decode
2021-06-10 21:13:38 +08:00
cuicheng01 d187ce20f3 Update legendary_models configs 2021-06-10 08:18:28 +00:00
cuicheng01 1ca24932e7 Update configs 2021-06-10 06:59:25 +00:00
cuicheng01 958727f39b update resnet condigs 2021-06-10 02:59:29 +00:00
littletomatodonkey c267c4a614 fix yaml 2021-06-08 05:29:37 +00:00
cuicheng01 36f4b36349 Update ImageNet configs 2021-06-08 04:11:48 +00:00
littletomatodonkey e5df00a02d
Update ResNet50.yaml 2021-06-08 11:50:45 +08:00
cuicheng01 56cd24552f
Update ResNet50.yaml 2021-06-02 19:03:12 +08:00
cuicheng01 1e8a020f2b fix bug of image transforms 2021-06-02 10:57:38 +00:00
cuicheng01 d13cb22378 Add legendary_model's congigs and fix some trainer's bug 2021-06-01 17:24:21 +00:00
Wei Shengyu aecbf40bed
Merge pull request #762 from Intsigstephon/develop_reg
add RecModel for retrieval
2021-06-01 14:43:29 +08:00
littletomatodonkey 0a0e34298b
add support for infer and export (#760)
* add support for infer

* add export model support

* fix yaml

* fix post process name

* fix topk name
2021-06-01 14:37:47 +08:00
Bin Lu 2eb2c26907
Update ResNet50_retrieval.yml 2021-06-01 14:24:48 +08:00
Bin Lu 3c184362a7
Create ResNet50_retrieval.yml 2021-06-01 14:23:40 +08:00
Bin Lu fa15e53b00
Update ResNet50.yaml 2021-06-01 14:23:07 +08:00
Bin Lu ea53ad95c3
Update ResNet50.yaml 2021-06-01 11:40:17 +08:00
littletomatodonkey 3881343484
add support for train and eval (#752)
* add support for train and eval

* rm unsed code

* add support for metric save and load ckp
2021-05-29 13:52:32 +08:00
littletomatodonkey dd79f81fd7
[WIP]add arch init (#744)
* polish trainer
2021-05-27 18:41:44 +08:00