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

stid config

This commit is contained in:
haoyu.he 2026-01-04 14:00:15 +08:00
parent c8416c5b01
commit 2bfa444f8e
13 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/STID/AirQuality.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

2
config/STID/BJTaxi-InFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

2
config/STID/BJTaxi-OutFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

2
config/STID/BJTaxi_InFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

2
config/STID/BJTaxi_OutFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

2
config/STID/METR-LA.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 300

2
config/STID/NYCBike-InFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

2
config/STID/NYCBike-OutFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

2
config/STID/NYCBike_InFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

2
config/STID/NYCBike_OutFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

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

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 300

2
config/STID/SolarEnergy.yaml
View File

@ -40,7 +40,7 @@ model:
train:
batch_size: 64
debug: true
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100

8
train.py
View File

@ -90,9 +90,9 @@ def read_config(config_path):
config = yaml.safe_load(file)
# 全局配置
device = "cuda:0" # 指定设备为cuda:0
device = "cuda:1" # 指定设备为cuda:0
seed = 2023 # 随机种子
epochs = 10 # 训练轮数
epochs = 100 # 训练轮数
# 拷贝项
config["basic"]["seed"] = seed
@ -121,5 +121,5 @@ if __name__ == "__main__":
all_dataset = big_dataset + mid_dataset + regular_dataset
dataset_list = regular_dataset
main(model_list, dataset_list, debug=True)
dataset_list = all_dataset
main(model_list, dataset_list, debug=False)