124 Commits

Author SHA1 Message Date
jm12138
c40ce17aca update the configs of models 2021-06-10 23:40:29 +08:00
jm12138
33e2c673b5 add the configs of models 2021-06-10 23:35:50 +08:00
littletomatodonkey
f4f0984029
Update mv3_large_x1_0_distill_mv3_small_x1_0.yaml 2021-06-10 21:41:07 +08:00
littletomatodonkey
63cf3ba4d1
fix imread (#824)
* fix imread

* rm unused decode

* rm decode
2021-06-10 21:13:38 +08:00
littletomatodonkey
3b4f5f4dfc
add distillation and fix some apis (#810)
* fix save load and imagenet dataset
* refine trainer
2021-06-10 16:30:05 +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
9ad588e048
Update HRNet_W48_C.yaml 2021-06-02 19:08:00 +08:00
cuicheng01
56cd24552f
Update ResNet50.yaml 2021-06-02 19:03:12 +08:00
cuicheng01
dc44df5a21
Update MobileNetV3_large_x1_0.yaml 2021-06-02 19:02:28 +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