mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-11-07 21:33:27 +00:00
Merge remote-tracking branch 'origin/dygraph' into dygraph
This commit is contained in:
commit
69df224ce5
@ -82,7 +82,7 @@ Train:
|
|||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: LMDBDataSet
|
name: LMDBDataSet
|
||||||
data_dir: ./train_data/data_lmdb_release/validation/
|
data_dir: ./train_data/data_lmdb_release/evaluaiton/
|
||||||
transforms:
|
transforms:
|
||||||
- DecodeImage: # load image
|
- DecodeImage: # load image
|
||||||
img_mode: BGR
|
img_mode: BGR
|
||||||
|
|||||||
@ -82,7 +82,7 @@ Train:
|
|||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: LMDBDataSet
|
name: LMDBDataSet
|
||||||
data_dir: ./train_data/data_lmdb_release/validation/
|
data_dir: ./train_data/data_lmdb_release/evaluation/
|
||||||
transforms:
|
transforms:
|
||||||
- DecodeImage: # load image
|
- DecodeImage: # load image
|
||||||
img_mode: RGB
|
img_mode: RGB
|
||||||
|
|||||||
@ -77,7 +77,7 @@ Metric:
|
|||||||
Train:
|
Train:
|
||||||
dataset:
|
dataset:
|
||||||
name: LMDBDataSet
|
name: LMDBDataSet
|
||||||
data_dir: ./train_data/data_lmdb_release/training
|
data_dir: ./train_data/data_lmdb_release/training/
|
||||||
transforms:
|
transforms:
|
||||||
- DecodeImage: # load image
|
- DecodeImage: # load image
|
||||||
img_mode: BGR
|
img_mode: BGR
|
||||||
@ -97,7 +97,7 @@ Train:
|
|||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: LMDBDataSet
|
name: LMDBDataSet
|
||||||
data_dir: ./train_data/data_lmdb_release/validation
|
data_dir: ./train_data/data_lmdb_release/evaluation/
|
||||||
transforms:
|
transforms:
|
||||||
- DecodeImage: # load image
|
- DecodeImage: # load image
|
||||||
img_mode: BGR
|
img_mode: BGR
|
||||||
|
|||||||
@ -81,7 +81,7 @@ Train:
|
|||||||
Eval:
|
Eval:
|
||||||
dataset:
|
dataset:
|
||||||
name: LMDBDataSet
|
name: LMDBDataSet
|
||||||
data_dir: ./train_data/data_lmdb_release/validation/
|
data_dir: ./train_data/data_lmdb_release/evaluaiton/
|
||||||
transforms:
|
transforms:
|
||||||
- DecodeImage: # load image
|
- DecodeImage: # load image
|
||||||
img_mode: BGR
|
img_mode: BGR
|
||||||
|
|||||||
@ -273,7 +273,8 @@ def _get_length(logit):
|
|||||||
out = out.cast('int32')
|
out = out.cast('int32')
|
||||||
out = out.argmax(-1)
|
out = out.argmax(-1)
|
||||||
out = out + 1
|
out = out + 1
|
||||||
out = paddle.where(abn, out, paddle.to_tensor(logit.shape[1]))
|
len_seq = paddle.zeros_like(out) + logit.shape[1]
|
||||||
|
out = paddle.where(abn, out, len_seq)
|
||||||
return out
|
return out
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@ -51,9 +51,3 @@ null:null
|
|||||||
null:null
|
null:null
|
||||||
===========================infer_benchmark_params==========================
|
===========================infer_benchmark_params==========================
|
||||||
random_infer_input:[{float32,[3,640,640]}];[{float32,[3,960,960]}]
|
random_infer_input:[{float32,[3,640,640]}];[{float32,[3,960,960]}]
|
||||||
===========================train_benchmark_params==========================
|
|
||||||
batch_size:8|16
|
|
||||||
fp_items:fp32|fp16
|
|
||||||
epoch:2
|
|
||||||
--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
|
|
||||||
flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096
|
|
||||||
|
|||||||
@ -9,7 +9,7 @@
|
|||||||
|
|
||||||
```shell
|
```shell
|
||||||
# 运行格式:bash test_tipc/prepare.sh train_benchmark.txt mode
|
# 运行格式:bash test_tipc/prepare.sh train_benchmark.txt mode
|
||||||
bash test_tipc/prepare.sh test_tipc/configs/det_mv3_db_v2_0/train_benchmark.txt benchmark_train
|
bash test_tipc/prepare.sh test_tipc/configs/det_mv3_db_v2_0/train_infer_python.txt benchmark_train
|
||||||
```
|
```
|
||||||
|
|
||||||
## 1.2 功能测试
|
## 1.2 功能测试
|
||||||
@ -33,7 +33,7 @@ dynamic_bs8_fp32_DP_N1C1为test_tipc/benchmark_train.sh传入的参数,格式
|
|||||||
|
|
||||||
## 2. 日志输出
|
## 2. 日志输出
|
||||||
|
|
||||||
运行后将保存模型的训练日志和解析日志,使用 `test_tipc/configs/det_mv3_db_v2_0/train_benchmark.txt` 参数文件的训练日志解析结果是:
|
运行后将保存模型的训练日志和解析日志,使用 `test_tipc/configs/det_mv3_db_v2_0/train_infer_python.txt` 参数文件的训练日志解析结果是:
|
||||||
|
|
||||||
```
|
```
|
||||||
{"model_branch": "dygaph", "model_commit": "7c39a1996b19087737c05d883fd346d2f39dbcc0", "model_name": "det_mv3_db_v2_0_bs8_fp32_SingleP_DP", "batch_size": 8, "fp_item": "fp32", "run_process_type": "SingleP", "run_mode": "DP", "convergence_value": "5.413110", "convergence_key": "loss:", "ips": 19.333, "speed_unit": "samples/s", "device_num": "N1C1", "model_run_time": "0", "frame_commit": "8cc09552473b842c651ead3b9848d41827a3dbab", "frame_version": "0.0.0"}
|
{"model_branch": "dygaph", "model_commit": "7c39a1996b19087737c05d883fd346d2f39dbcc0", "model_name": "det_mv3_db_v2_0_bs8_fp32_SingleP_DP", "batch_size": 8, "fp_item": "fp32", "run_process_type": "SingleP", "run_mode": "DP", "convergence_value": "5.413110", "convergence_key": "loss:", "ips": 19.333, "speed_unit": "samples/s", "device_num": "N1C1", "model_run_time": "0", "frame_commit": "8cc09552473b842c651ead3b9848d41827a3dbab", "frame_version": "0.0.0"}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user