diff --git a/doc/doc_ch/tree.md b/doc/doc_ch/tree.md
index b31321462733da66b65beba32033055418cc6927..511860869188020ccfe7531db55ecc5008ccb56b 100644
--- a/doc/doc_ch/tree.md
+++ b/doc/doc_ch/tree.md
@@ -135,7 +135,8 @@ PaddleOCR
 │   │   ├── det_db_loss.py                  // DB loss
 │   │   ├── det_east_loss.py                // EAST loss
 │   │   ├── det_sast_loss.py                // SAST loss
-│   │   ├── rec_ctc_loss.py                 // ctc loss
+│   │   ├── rec_ctc_loss.py                 // CTC loss
+│   │   ├── rec_att_loss.py                 // Attention loss
 │   ├── metrics                             // 评估指标
 │   │   ├── __init__.py                     // 构造 metric 相关代码
 │   │   ├── cls_metric.py                   // 方向分类器 metric
@@ -162,10 +163,11 @@ PaddleOCR
 │   │   ├── heads                           // 头函数
 │   │   │   ├── __init__.py                 // 构造 head 相关代码
 │   │   │   ├── cls_head.py                 // 方向分类器 分类头
-│   │   │   ├── det_db_head.py              // db 检测头
+│   │   │   ├── det_db_head.py              // DB 检测头
 │   │   │   ├── det_east_head.py            // EAST 检测头
 │   │   │   ├── det_sast_head.py            // SAST 检测头
 │   │   │   ├── rec_ctc_head.py             // 识别 ctc
+│   │   │   ├── rec_att_head.py             // 识别 attention 
 │   │   ├── transforms                      // 图像变换
 │   │   │   ├── __init__.py                 // 构造 transform 相关代码
 │   │   │   └── tps.py                      // TPS 变换
diff --git a/doc/doc_en/tree_en.md b/doc/doc_en/tree_en.md
index 461bb1848d475a15f97c35b4a76dd487f364b643..be2f5733d3c30af8c83f1072cd0aea07865bacc3 100644
--- a/doc/doc_en/tree_en.md
+++ b/doc/doc_en/tree_en.md
@@ -138,7 +138,8 @@ PaddleOCR
 │   │   ├── det_db_loss.py                  // DB loss
 │   │   ├── det_east_loss.py                // EAST loss
 │   │   ├── det_sast_loss.py                // SAST loss
-│   │   ├── rec_ctc_loss.py                 // ctc loss
+│   │   ├── rec_ctc_loss.py                 // CTC loss
+│   │   ├── rec_att_loss.py                 // Attention loss
 │   ├── metrics                             // Metrics
 │   │   ├── __init__.py                     // Construct metric code
 │   │   ├── cls_metric.py                   // Angle class metric
@@ -168,7 +169,8 @@ PaddleOCR
 │   │   │   ├── det_db_head.py              // DB head
 │   │   │   ├── det_east_head.py            // EAST head
 │   │   │   ├── det_sast_head.py            // SAST head
-│   │   │   ├── rec_ctc_head.py             // Ctc head
+│   │   │   ├── rec_ctc_head.py             // CTC head
+│   │   │   ├── rec_att_head.py             // Attention head
 │   │   ├── transforms                      // Transforms
 │   │   │   ├── __init__.py                 // Construct transform code
 │   │   │   └── tps.py                      // TPS transform