From e6121a317345fea4d7245ec4eb0516a5498967da Mon Sep 17 00:00:00 2001 From: majorli6 Date: Mon, 21 Nov 2022 11:37:31 +0800 Subject: [PATCH] add training results for satrn model link #I5Y013 Signed-off-by: majorli6 --- cv/ocr/satrn/pytorch/base/README.md | 6 ++++++ cv/ocr/satrn/pytorch/base/configs/datasets_/ST_MJ_train.py | 4 ++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/cv/ocr/satrn/pytorch/base/README.md b/cv/ocr/satrn/pytorch/base/README.md index 20396410a..0cc6dfec2 100755 --- a/cv/ocr/satrn/pytorch/base/README.md +++ b/cv/ocr/satrn/pytorch/base/README.md @@ -115,6 +115,12 @@ $ bash dist_train.sh configs/models/satrn_academic.py 8 | :-----: |:-------:| :-------: |:--------: | | satrn | BI100x8 | 14.159G | 549.94 | +| dataset | acc | +| :-----: |:-------:| +| IIIT5K | 94.5 | +| IC15 | 83.3 | +| SVTP | 88.4 | + | Convergence criteria | Configuration (x denotes number of GPUs) | Performance | Accuracy | Power（W） | Scalability | Memory utilization（G） | Stability | |----------------------|------------------------------------------|-------------|----------|------------|-------------|-------------------------|-----------| | 0.841 | SDK V2.2,bs:128,8x,fp32 | 630 | N/A | 166\*8 | 0.98 | 28.5\*8 | 1 | diff --git a/cv/ocr/satrn/pytorch/base/configs/datasets_/ST_MJ_train.py b/cv/ocr/satrn/pytorch/base/configs/datasets_/ST_MJ_train.py index 0bd1490af..e0873d604 100755 --- a/cv/ocr/satrn/pytorch/base/configs/datasets_/ST_MJ_train.py +++ b/cv/ocr/satrn/pytorch/base/configs/datasets_/ST_MJ_train.py @@ -29,5 +29,5 @@ train2 = {key: value for key, value in train1.items()} train2['img_prefix'] = train_img_prefix2 train2['ann_file'] = train_ann_file2 -# train_list = [train1, train2] -train_list = [train1] \ No newline at end of file +train_list = [train1, train2] +# train_list = [train1] \ No newline at end of file -- Gitee