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 |