diff --git a/tools/train.py b/tools/train.py
index 0f5e9039cfd4cc23e418434323b74c6612587ed2..68e792b7331a9d47ca6744ea1a9f362979d75542 100755
--- a/tools/train.py
+++ b/tools/train.py
@@ -74,7 +74,7 @@ def main():
 
     # dump mode structure
     if config['Global']['debug']:
-        if 'attention' in config['Global']['loss_type']:
+        if train_alg_type == 'rec' and 'attention' in config['Global']['loss_type']:
             logger.warning('Does not suport dump attention...')
         else:
             summary(train_program)