From 0a92617cb082eca60c3de590fc2307994cf1a687 Mon Sep 17 00:00:00 2001
From: tink2123 <y_tink@163.com>
Date: Mon, 13 Jul 2020 11:22:23 +0800
Subject: [PATCH] update readme for space and add thanks for user

---
 README.md      | 3 ++-
 README_en.md   | 3 ++-
 tools/train.py | 2 +-
 3 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 283c74f6..11c83ea1 100644
--- a/README.md
+++ b/README.md
@@ -4,7 +4,7 @@
 PaddleOCR旨在打造一套丰富、领先、且实用的OCR工具库，助力使用者训练出更好的模型，并应用落地。
 
 **近期更新**
-- 2020.7.9 添加支持空格的识别模型，[识别效果](#支持空格的中文OCR效果展示)
+- 2020.7.9 添加支持空格的识别模型，[识别效果](#支持空格的中文OCR效果展示)，[训练](./doc/doc_ch/recognition.md)及[预测](./doc/doc_ch/quickstart.md)
 - 2020.7.9 添加数据增强、学习率衰减策略,具体参考[配置文件](./doc/doc_ch/config.md)
 - 2020.6.8 添加[数据集](./doc/doc_ch/datasets.md)，并保持持续更新
 - 2020.6.5 支持 `attetnion` 模型导出 `inference_model`
@@ -179,3 +179,4 @@ PaddleOCR文本识别算法的训练和使用请参考文档教程中[模型训
 
 - 非常感谢 [Khanh Tran](https://github.com/xxxpsyduck) 贡献了英文文档。
 - 非常感谢 [zhangxin](https://github.com/ZhangXinNan)([Blog](https://blog.csdn.net/sdlypyzq)) 贡献新的可视化方式、添加.gitgnore、处理手动设置PYTHONPATH环境变量的问题
+- 非常感谢 [lyl120117](https://github.com/lyl120117) 贡献打印网络结构的代码
diff --git a/README_en.md b/README_en.md
index 24653eb5..671d3bc7 100644
--- a/README_en.md
+++ b/README_en.md
@@ -4,7 +4,7 @@ English | [简体中文](README.md)
 PaddleOCR aims to create a rich, leading, and practical OCR tools that help users train better models and apply them into practice.
 
 **Recent updates**、
-- 2020.7.9 Add recognition model to support space, [recognition result](#space Chinese OCR results)
+- 2020.7.9 Add recognition model to support space, [recognition result](#space Chinese OCR results),[Training](./doc/doc_ch/recognition.md)及[Inference](./doc/doc_ch/quickstart.md)
 - 2020.7.9 Add data auguments and learning rate decay strategies,please read [config](./doc/doc_en/config_en.md)
 - 2020.6.8 Add [dataset](./doc/doc_en/datasets_en.md) and keep updating
 - 2020.6.5 Support exporting `attention` model to `inference_model`
@@ -299,3 +299,4 @@ We welcome all the contributions to PaddleOCR and appreciate for your feedback v
 
 - Many thanks to [Khanh Tran](https://github.com/xxxpsyduck) for contributing the English documentation.
 - Many thanks to [zhangxin](https://github.com/ZhangXinNan) for contributing the new visualize function、add .gitgnore and discard set PYTHONPATH manually.
+- Many thanks to [lyl120117](https://github.com/lyl120117) for contributing the code for printing the network structure.
diff --git a/tools/train.py b/tools/train.py
index c8350ff6..ccdf3c70 100755
--- a/tools/train.py
+++ b/tools/train.py
@@ -91,7 +91,7 @@ def main():
 
     # dump mode structure
     if config['Global']['debug']:
-        if 'Attention' in config['Head'].keys():
+        if 'attention' in config['Global']['loss_type']:
             logger.warning('Does not suport dump attention...')
         else:
             summary(train_program)
-- 
GitLab