diff --git a/config/Informer/BJTaxi-InFlow.yaml b/config/Informer/BJTaxi-InFlow.yaml
index effda89..538edbe 100644
--- a/config/Informer/BJTaxi-InFlow.yaml
+++ b/config/Informer/BJTaxi-InFlow.yaml
@@ -59,7 +59,7 @@ train:
   mae_thresh: None
   mape_thresh: 0.001
   max_grad_norm: 5
-  output_dim: 6
+  output_dim: 1
   plot: false
   pred_len: 24
   real_value: true