diff --git a/docs/zh_CN/models/TNT.md b/docs/zh_CN/models/TNT.md
index 18cd490ceca8af52005a69d827cc9b6921465bd6..13e86fd4c1dd5b9710ba3a81580d742de95915aa 100644
--- a/docs/zh_CN/models/TNT.md
+++ b/docs/zh_CN/models/TNT.md
@@ -9,4 +9,4 @@ TNT（Transformer-iN-Transformer）系列模型由华为诺亚于2021年提出
 
 |         Model        | Params (M) | FLOPs (G) | Top-1 (%) | Top-5 (%) |
 |:---------------------:|:----------:|:---------:|:---------:|:---------:|
-|        TNT_small        | 23.8       | 5.2       | 81.12     |   95.56   |
\ No newline at end of file
+|        TNT_small        | 23.8       | 5.2       | 81.21     |   95.63   |
diff --git a/ppcls/data/__init__.py b/ppcls/data/__init__.py
index 3bbf75afa7c9d1c61351e20af6d7bdf6cf50818b..0804d3d49330b2de175e2b9f85bcf174bf283a0d 100644
--- a/ppcls/data/__init__.py
+++ b/ppcls/data/__init__.py
@@ -58,7 +58,9 @@ def build_dataloader(config, mode, device, seed=None):
         'Train',
         'Eval',
         'Test',
-    ], "Mode should be Train, Eval, Test"
+        'Gallery',
+        'Query'
+    ], "Mode should be Train, Eval, Test, Gallery, Query"
     # build dataset
     config_dataset = config[mode]['dataset']
     config_dataset = copy.deepcopy(config_dataset)