diff --git a/README.md b/README.md
index 283c74f662d08daf6b361c64a11a90a690e9227f..11c83ea157608a72221a0d0a5b64766e407af584 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 24653eb5c929faa130cd90cba6a328ab8171bdcf..671d3bc7893d7a7107bb18a6a26743fac22c42a3 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 c8350ff64b4894cb22bde063529786b0945dfea3..ccdf3c70907eca311407b81548da653cbb1b987f 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)