diff --git a/config/REPST/PEMS-BAY.yaml b/config/REPST/PEMS-BAY.yaml
index 6fcf12b..33e582e 100755
--- a/config/REPST/PEMS-BAY.yaml
+++ b/config/REPST/PEMS-BAY.yaml
@@ -34,7 +34,7 @@ model:
   d_model: 64
   n_heads: 1
   input_dim: 1
-  word_num: 2000
+  word_num: 1000
 
 train:
   batch_size: 16