czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

改word_num为1000

This commit is contained in:
czzhangheng 2025-11-12 17:21:56 +08:00
parent e61706c391
commit dceae4b1a3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/REPST/PEMS-BAY.yaml
View File

@ -34,7 +34,7 @@ model:
d_model: 64 d_model: 64
n_heads: 1 n_heads: 1
input_dim: 1 input_dim: 1
word_num: 2000 word_num: 1000
train: train:
batch_size: 16 batch_size: 16