TrafficWheel/test_results.txt

5407 lines
231 KiB
Plaintext

# 测试报告
## 测试概述
- 测试时间: 2025-12-01 22:20:35
- 总测试文件数: 252
- 通过: 41
- 失败: 0
- 错误: 211
## 通过的配置文件
- ✅ /user/czzhangheng/code/TrafficWheel/config/STID/NYCBike_Inflow.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STID/METR-LA.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STID/AirQuality.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STID/PEMS-BAY.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STID/NYCBike_Outflow.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STID/SolarEnergy.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STAEFormer/PEMSD4.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STAEFormer/METR-LA.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STAEFormer/PEMSD8.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STAEFormer/PEMSD3.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/STAEFormer/PEMSD7.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/AEPSA/NYCBike-inflow.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/AEPSA/METR-LA.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/AEPSA/AirQuality.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/AEPSA/PEMS-BAY.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/AEPSA/SolarEnergy.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/AEPSA/v2_METR-LA.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/AEPSA/NYCBike-outflow.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/AEPSA/v2_SolarEnergy.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/TCN/NYCBike-InFlow.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/TCN/PEMSD4.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/TCN/METR-LA.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/TCN/NYCBike-OutFlow.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/TCN/PEMSD8.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/TCN/PEMSD3.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/TCN/SolarEnergy.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/TCN/PEMSD7.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/METR-LA.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD8.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/DSANET/PEMSD4.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/DSANET/METR-LA.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/DSANET/PEMSD8.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/DSANET/PEMSD3.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/DSANET/PEMSD7.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/REPST/NYCBike-inflow.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/REPST/BeijingAirQuality(Deprecated).yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/REPST/METR-LA.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/REPST/AirQuality.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/REPST/PEMS-BAY.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/REPST/SolarEnergy.yaml
- ✅ /user/czzhangheng/code/TrafficWheel/config/REPST/NYCBike-outflow.yaml
## 失败的配置文件
## 出错的配置文件
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXPB/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXPB/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXPB/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXPB/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXPB/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXPB/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD7(L).yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/Hainan.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD7(M).yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STSGCN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STSGCN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STSGCN/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STSGCN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STSGCN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STSGCN/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STSGCN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STSGCN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STSGCN/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STID/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STID/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STID/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAWnet/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAWnet/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAWnet/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAWnet/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAWnet/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAWnet/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAWnet/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAWnet/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAWnet/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DCRNN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DCRNN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DCRNN/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DCRNN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DCRNN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DCRNN/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DCRNN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DCRNN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DCRNN/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAEFormer/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAEFormer/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAEFormer/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STAEFormer/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGODE/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGODE/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGODE/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGODE/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGODE/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGODE/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGODE/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGODE/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGODE/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNCDE/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNCDE/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNCDE/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNCDE/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNCDE/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNCDE/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNCDE/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNCDE/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNCDE/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AEPSA/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AEPSA/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ST_SSL/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ST_SSL/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ST_SSL/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ST_SSL/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ST_SSL/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ST_SSL/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ST_SSL/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ST_SSL/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ST_SSL/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/TCN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/SD.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/EXP/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD7(L).yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/Hainan.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD7(M).yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DSANET/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DSANET/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DSANET/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/DSANET/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STFGNN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STFGNN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STFGNN/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STFGNN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STFGNN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STFGNN/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STFGNN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STFGNN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STFGNN/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AGCRN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AGCRN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AGCRN/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AGCRN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AGCRN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AGCRN/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AGCRN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AGCRN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/AGCRN/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNRDE/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNRDE/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNRDE/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNRDE/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNRDE/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNRDE/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNRDE/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNRDE/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGNRDE/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/REPST/PEMS-BAY_paper.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/REPST/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/REPST/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STIDGCN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STIDGCN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STIDGCN/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STIDGCN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STIDGCN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STIDGCN/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STIDGCN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STIDGCN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STIDGCN/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/NLT/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/NLT/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/NLT/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/NLT/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/NLT/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/NLT/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/NLT/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/NLT/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/NLT/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD7(L).yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/Hainan.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD7(M).yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STMLP/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STMLP/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STMLP/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STMLP/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STMLP/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STMLP/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STMLP/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STMLP/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STMLP/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/MegaCRN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/MegaCRN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/MegaCRN/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/MegaCRN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/MegaCRN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/MegaCRN/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/MegaCRN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/MegaCRN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/MegaCRN/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/GWN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/GWN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/GWN/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/GWN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/GWN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/GWN/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/GWN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/GWN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/GWN/PEMSD7.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGCN/NYCBike-InFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGCN/PEMSD4.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGCN/METR-LA.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGCN/AirQuality.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGCN/NYCBike-OutFlow.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGCN/PEMSD8.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGCN/PEMSD3.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGCN/SolarEnergy.yaml
- ⚠️ /user/czzhangheng/code/TrafficWheel/config/STGCN/PEMSD7.yaml
## 详细输出
### PASSED
#### /user/czzhangheng/code/TrafficWheel/config/STID/NYCBike_Inflow.yaml
```
模型参数量: 118040
加载 NYCBike-InFlow 数据集中...
✓ Test passed: output shape torch.Size([64, 24, 128, 1]) matches label shape torch.Size([64, 24, 128, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STID/METR-LA.yaml
```
模型参数量: 120568
加载 METR-LA 数据集中...
✓ Test passed: output shape torch.Size([64, 24, 207, 1]) matches label shape torch.Size([64, 24, 207, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STID/AirQuality.yaml
```
模型参数量: 115064
加载 AirQuality 数据集中...
✓ Test passed: output shape torch.Size([64, 24, 35, 1]) matches label shape torch.Size([64, 24, 35, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STID/PEMS-BAY.yaml
```
模型参数量: 124344
加载 PEMS-BAY 数据集中...
✓ Test passed: output shape torch.Size([64, 24, 325, 1]) matches label shape torch.Size([64, 24, 325, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STID/NYCBike_Outflow.yaml
```
模型参数量: 118040
加载 NYCBike-OutFlow 数据集中...
✓ Test passed: output shape torch.Size([64, 24, 128, 1]) matches label shape torch.Size([64, 24, 128, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STID/SolarEnergy.yaml
```
模型参数量: 118328
加载 SolarEnergy 数据集中...
✓ Test passed: output shape torch.Size([64, 24, 137, 1]) matches label shape torch.Size([64, 24, 137, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STAEFormer/PEMSD4.yaml
```
模型参数量: 1354932
加载 PEMSD4 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/STAEFormer/PEMSD4/2025-12-01_21-52-10/run.log
✓ Test passed: output shape torch.Size([64, 12, 307, 1]) matches label shape torch.Size([64, 12, 307, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STAEFormer/METR-LA.yaml
```
模型参数量: 1258932
加载 METR-LA 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/STAEFormer/METR-LA/2025-12-01_21-52-24/run.log
✓ Test passed: output shape torch.Size([16, 12, 207, 1]) matches label shape torch.Size([16, 12, 207, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STAEFormer/PEMSD8.yaml
```
模型参数量: 1223412
加载 PEMSD8 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/STAEFormer/PEMSD8/2025-12-01_21-52-49/run.log
✓ Test passed: output shape torch.Size([64, 12, 170, 1]) matches label shape torch.Size([64, 12, 170, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STAEFormer/PEMSD3.yaml
```
模型参数量: 1403892
加载 PEMSD3 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/STAEFormer/PEMSD3/2025-12-01_21-53-06/run.log
✓ Test passed: output shape torch.Size([16, 12, 358, 1]) matches label shape torch.Size([16, 12, 358, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/STAEFormer/PEMSD7.yaml
```
模型参数量: 1907892
加载 PEMSD7 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/STAEFormer/PEMSD7/2025-12-01_21-54-04/run.log
✓ Test passed: output shape torch.Size([16, 12, 883, 1]) matches label shape torch.Size([16, 12, 883, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/NYCBike-inflow.yaml
```
模型参数量: 103504579
加载 NYCBike-InFlow 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/AEPSA/NYCBike-InFlow/2025-12-01_21-55-58/run.log
✓ Test passed: output shape torch.Size([32, 24, 128, 1]) matches label shape torch.Size([32, 24, 128, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/METR-LA.yaml
```
模型参数量: 103505369
加载 METR-LA 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/AEPSA/METR-LA/2025-12-01_21-56-29/run.log
✓ Test passed: output shape torch.Size([16, 24, 207, 1]) matches label shape torch.Size([16, 24, 207, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/AirQuality.yaml
```
模型参数量: 103503669
加载 AirQuality 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/AEPSA/AirQuality/2025-12-01_21-56-40/run.log
✓ Test passed: output shape torch.Size([16, 24, 35, 6]) matches label shape torch.Size([16, 24, 35, 6])
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/PEMS-BAY.yaml
```
模型参数量: 103506549
加载 PEMS-BAY 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/AEPSA/PEMS-BAY/2025-12-01_21-57-30/run.log
✓ Test passed: output shape torch.Size([16, 24, 325, 1]) matches label shape torch.Size([16, 24, 325, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/SolarEnergy.yaml
```
模型参数量: 103504669
加载 SolarEnergy 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/AEPSA/SolarEnergy/2025-12-01_21-57-55/run.log
✓ Test passed: output shape torch.Size([64, 24, 137, 1]) matches label shape torch.Size([64, 24, 137, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/v2_METR-LA.yaml
```
模型参数量: 103524820
加载 METR-LA 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/AEPSA_v2/METR-LA/2025-12-01_21-58-18/run.log
✓ Test passed: output shape torch.Size([16, 24, 207, 1]) matches label shape torch.Size([16, 24, 207, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/NYCBike-outflow.yaml
```
模型参数量: 103504579
加载 NYCBike-OutFlow 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/AEPSA/NYCBike-OutFlow/2025-12-01_21-58-29/run.log
✓ Test passed: output shape torch.Size([32, 24, 128, 1]) matches label shape torch.Size([32, 24, 128, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/v2_SolarEnergy.yaml
```
模型参数量: 103524120
加载 SolarEnergy 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/AEPSA_v2/SolarEnergy/2025-12-01_21-58-54/run.log
✓ Test passed: output shape torch.Size([64, 24, 137, 1]) matches label shape torch.Size([64, 24, 137, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/TCN/NYCBike-InFlow.yaml
```
模型参数量: 35873
加载 NYCBike-InFlow 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/NYCBike-InFlow/2025-12-01_21-59-55/run.log
✓ Test passed: output shape torch.Size([32, 24, 128, 1]) matches label shape torch.Size([32, 24, 128, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/TCN/PEMSD4.yaml
```
模型参数量: 35873
加载 PEMSD4 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/PEMSD4/2025-12-01_22-00-07/run.log
✓ Test passed: output shape torch.Size([64, 12, 307, 1]) matches label shape torch.Size([64, 12, 307, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/TCN/METR-LA.yaml
```
模型参数量: 35873
加载 METR-LA 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/METR-LA/2025-12-01_22-00-28/run.log
✓ Test passed: output shape torch.Size([16, 24, 207, 1]) matches label shape torch.Size([16, 24, 207, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/TCN/NYCBike-OutFlow.yaml
```
模型参数量: 35873
加载 NYCBike-OutFlow 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/NYCBike-OutFlow/2025-12-01_22-00-44/run.log
✓ Test passed: output shape torch.Size([32, 24, 128, 1]) matches label shape torch.Size([32, 24, 128, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/TCN/PEMSD8.yaml
```
模型参数量: 35873
加载 PEMSD8 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/PEMSD8/2025-12-01_22-00-54/run.log
✓ Test passed: output shape torch.Size([64, 12, 170, 1]) matches label shape torch.Size([64, 12, 170, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/TCN/PEMSD3.yaml
```
模型参数量: 35873
加载 PEMSD3 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/PEMSD3/2025-12-01_22-01-10/run.log
✓ Test passed: output shape torch.Size([16, 12, 358, 1]) matches label shape torch.Size([16, 12, 358, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/TCN/SolarEnergy.yaml
```
模型参数量: 35873
加载 SolarEnergy 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/SolarEnergy/2025-12-01_22-01-33/run.log
✓ Test passed: output shape torch.Size([64, 24, 137, 1]) matches label shape torch.Size([64, 24, 137, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/TCN/PEMSD7.yaml
```
模型参数量: 35873
加载 PEMSD7 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/PEMSD7/2025-12-01_22-02-05/run.log
✓ Test passed: output shape torch.Size([16, 12, 883, 1]) matches label shape torch.Size([16, 12, 883, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/METR-LA.yaml
```
模型参数量: 671644
加载 METR-LA 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/DDGCRN/METR-LA/2025-12-01_22-03-35/run.log
✓ Test passed: output shape torch.Size([64, 24, 207, 1]) matches label shape torch.Size([64, 24, 207, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD8.yaml
```
模型参数量: 311759
加载 PEMSD8 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/DDGCRN/PEMSD8/2025-12-01_22-03-57/run.log
✓ Test passed: output shape torch.Size([64, 12, 170, 1]) matches label shape torch.Size([64, 12, 170, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/DSANET/PEMSD4.yaml
```
模型参数量: 37896712
加载 PEMSD4 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/DSANET/PEMSD4/2025-12-01_22-04-52/run.log
✓ Test passed: output shape torch.Size([64, 12, 307, 1]) matches label shape torch.Size([64, 12, 307, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/DSANET/METR-LA.yaml
```
模型参数量: 37896712
加载 METR-LA 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/DSANET/METR-LA/2025-12-01_22-05-06/run.log
✓ Test passed: output shape torch.Size([16, 12, 207, 1]) matches label shape torch.Size([16, 12, 207, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/DSANET/PEMSD8.yaml
```
模型参数量: 37896712
加载 PEMSD8 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/DSANET/PEMSD8/2025-12-01_22-05-33/run.log
✓ Test passed: output shape torch.Size([64, 12, 170, 1]) matches label shape torch.Size([64, 12, 170, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/DSANET/PEMSD3.yaml
```
模型参数量: 37896712
加载 PEMSD3 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/DSANET/PEMSD3/2025-12-01_22-05-49/run.log
✓ Test passed: output shape torch.Size([16, 12, 358, 1]) matches label shape torch.Size([16, 12, 358, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/DSANET/PEMSD7.yaml
```
模型参数量: 615304
加载 PEMSD7 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/DSANET/PEMSD7/2025-12-01_22-06-48/run.log
✓ Test passed: output shape torch.Size([16, 12, 883, 1]) matches label shape torch.Size([16, 12, 883, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/NYCBike-inflow.yaml
```
模型参数量: 103481647
加载 NYCBike-InFlow 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/REPST/NYCBike-InFlow/2025-12-01_22-09-34/run.log
✓ Test passed: output shape torch.Size([16, 24, 128, 1]) matches label shape torch.Size([16, 24, 128, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/BeijingAirQuality(Deprecated).yaml
```
模型参数量: 103815937
加载 BeijingAirQuality 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/REPST/BeijingAirQuality/2025-12-01_22-09-59/run.log
✓ Test passed: output shape torch.Size([16, 24, 7, 3]) matches label shape torch.Size([16, 24, 7, 3])
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/METR-LA.yaml
```
模型参数量: 103481647
加载 METR-LA 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/REPST/METR-LA/2025-12-01_22-10-22/run.log
✓ Test passed: output shape torch.Size([16, 24, 207, 1]) matches label shape torch.Size([16, 24, 207, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/AirQuality.yaml
```
模型参数量: 103815973
加载 AirQuality 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/REPST/AirQuality/2025-12-01_22-10-33/run.log
✓ Test passed: output shape torch.Size([16, 24, 35, 3]) matches label shape torch.Size([16, 24, 35, 3])
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/PEMS-BAY.yaml
```
模型参数量: 103481647
加载 PEMS-BAY 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/REPST/PEMS-BAY/2025-12-01_22-11-23/run.log
✓ Test passed: output shape torch.Size([16, 24, 325, 1]) matches label shape torch.Size([16, 24, 325, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/SolarEnergy.yaml
```
模型参数量: 103481647
加载 SolarEnergy 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/REPST/SolarEnergy/2025-12-01_22-11-48/run.log
✓ Test passed: output shape torch.Size([64, 24, 137, 1]) matches label shape torch.Size([64, 24, 137, 1])
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/NYCBike-outflow.yaml
```
模型参数量: 103481647
加载 NYCBike-OutFlow 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/REPST/NYCBike-OutFlow/2025-12-01_22-11-58/run.log
✓ Test passed: output shape torch.Size([16, 24, 128, 1]) matches label shape torch.Size([16, 24, 128, 1])
```
### FAILED
### ERROR
#### /user/czzhangheng/code/TrafficWheel/config/EXPB/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'to'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXPB/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'to'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXPB/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'to'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXPB/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'to'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXPB/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'to'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXPB/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'to'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 60, in model_selector
return STSGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/STSGCN.py", line 295, in __init__
self.adj = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/get_adj.py", line 10, in get_adj
match args["num_nodes"]:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 60, in model_selector
return STSGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/STSGCN.py", line 295, in __init__
self.adj = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/get_adj.py", line 10, in get_adj
match args["num_nodes"]:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 60, in model_selector
return STSGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/STSGCN.py", line 295, in __init__
self.adj = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/get_adj.py", line 10, in get_adj
match args["num_nodes"]:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 60, in model_selector
return STSGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/STSGCN.py", line 295, in __init__
self.adj = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/get_adj.py", line 10, in get_adj
match args["num_nodes"]:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 60, in model_selector
return STSGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/STSGCN.py", line 295, in __init__
self.adj = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/get_adj.py", line 10, in get_adj
match args["num_nodes"]:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 60, in model_selector
return STSGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/STSGCN.py", line 295, in __init__
self.adj = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/get_adj.py", line 10, in get_adj
match args["num_nodes"]:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 60, in model_selector
return STSGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/STSGCN.py", line 295, in __init__
self.adj = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/get_adj.py", line 10, in get_adj
match args["num_nodes"]:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 60, in model_selector
return STSGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/STSGCN.py", line 295, in __init__
self.adj = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/get_adj.py", line 10, in get_adj
match args["num_nodes"]:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 60, in model_selector
return STSGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/STSGCN.py", line 295, in __init__
self.adj = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STSGCN/get_adj.py", line 10, in get_adj
match args["num_nodes"]:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STID/NYCBike-InFlow.yaml
```
模型参数量: 146712
加载 NYCBike-InFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/STID/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 70, in model_selector
return STID(model_config)
^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STID/STID.py", line 13, in __init__
self.embed_dim = model_args["embed_dim"]
~~~~~~~~~~^^^^^^^^^^^^^
KeyError: 'embed_dim'
```
#### /user/czzhangheng/code/TrafficWheel/config/STID/NYCBike-OutFlow.yaml
```
模型参数量: 146712
加载 NYCBike-OutFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/STAWnet/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 82, in model_selector
return STAWnet(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAWnet/STAWnet.py", line 269, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STAWnet/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 82, in model_selector
return STAWnet(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAWnet/STAWnet.py", line 269, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STAWnet/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 82, in model_selector
return STAWnet(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAWnet/STAWnet.py", line 269, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STAWnet/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 82, in model_selector
return STAWnet(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAWnet/STAWnet.py", line 269, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STAWnet/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 82, in model_selector
return STAWnet(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAWnet/STAWnet.py", line 269, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STAWnet/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 82, in model_selector
return STAWnet(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAWnet/STAWnet.py", line 269, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STAWnet/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 82, in model_selector
return STAWnet(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAWnet/STAWnet.py", line 269, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STAWnet/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 82, in model_selector
return STAWnet(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAWnet/STAWnet.py", line 269, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STAWnet/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 82, in model_selector
return STAWnet(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAWnet/STAWnet.py", line 269, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DCRNN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 50, in model_selector
return DCRNNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DCRNN/dcrnn_model.py", line 123, in __init__
adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DCRNN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 50, in model_selector
return DCRNNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DCRNN/dcrnn_model.py", line 123, in __init__
adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DCRNN/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 50, in model_selector
return DCRNNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DCRNN/dcrnn_model.py", line 123, in __init__
adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DCRNN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 50, in model_selector
return DCRNNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DCRNN/dcrnn_model.py", line 123, in __init__
adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DCRNN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 50, in model_selector
return DCRNNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DCRNN/dcrnn_model.py", line 123, in __init__
adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DCRNN/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 50, in model_selector
return DCRNNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DCRNN/dcrnn_model.py", line 123, in __init__
adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DCRNN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 50, in model_selector
return DCRNNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DCRNN/dcrnn_model.py", line 123, in __init__
adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DCRNN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 50, in model_selector
return DCRNNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DCRNN/dcrnn_model.py", line 123, in __init__
adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 29, in get_adj
return adj
^^^
UnboundLocalError: cannot access local variable 'adj' where it is not associated with a value
```
#### /user/czzhangheng/code/TrafficWheel/config/DCRNN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 50, in model_selector
return DCRNNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DCRNN/dcrnn_model.py", line 123, in __init__
adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STAEFormer/NYCBike-InFlow.yaml
```
模型参数量: 3086208
加载 NYCBike-InFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/STAEFormer/AirQuality.yaml
```
模型参数量: 1624752
加载 AirQuality 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/STAEFormer/AirQuality/2025-12-01_21-52-33/run.log
2025/12/01 21:52:33 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/STAEFormer/AirQuality/2025-12-01_21-52-33
2025/12/01 21:52:33 - Training process started
Train Epoch 1: 0%| | 0/325 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/325 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAEFormer/STAEFormer.py", line 195, in forward
x = self.input_proj(x) # (batch_size, in_steps, num_nodes, input_embedding_dim)
^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/linear.py", line 125, in forward
return F.linear(input, self.weight, self.bias)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: mat1 and mat2 shapes cannot be multiplied (13440x1 and 6x24)
```
#### /user/czzhangheng/code/TrafficWheel/config/STAEFormer/NYCBike-OutFlow.yaml
```
模型参数量: 3086208
加载 NYCBike-OutFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/STAEFormer/SolarEnergy.yaml
```
模型参数量: 13296192
加载 SolarEnergy 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/STAEFormer/SolarEnergy/2025-12-01_21-53-32/run.log
2025/12/01 21:53:32 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/STAEFormer/SolarEnergy/2025-12-01_21-53-32
2025/12/01 21:53:32 - Training process started
Train Epoch 1: 0%| | 0/1970 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/1970 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STAEFormer/STAEFormer.py", line 195, in forward
x = self.input_proj(x) # (batch_size, in_steps, num_nodes, input_embedding_dim)
^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/linear.py", line 125, in forward
return F.linear(input, self.weight, self.bias)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: mat1 and mat2 shapes cannot be multiplied (52608x1 and 137x24)
```
#### /user/czzhangheng/code/TrafficWheel/config/STGODE/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 62, in model_selector
return ODEGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGODE/STGODE.py", line 149, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGODE/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 62, in model_selector
return ODEGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGODE/STGODE.py", line 149, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGODE/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 62, in model_selector
return ODEGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGODE/STGODE.py", line 149, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGODE/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 62, in model_selector
return ODEGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGODE/STGODE.py", line 149, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGODE/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 62, in model_selector
return ODEGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGODE/STGODE.py", line 149, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGODE/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 62, in model_selector
return ODEGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGODE/STGODE.py", line 149, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGODE/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 62, in model_selector
return ODEGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGODE/STGODE.py", line 149, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGODE/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 62, in model_selector
return ODEGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGODE/STGODE.py", line 149, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGODE/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 62, in model_selector
return ODEGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGODE/STGODE.py", line 149, in __init__
num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNCDE/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 44, in model_selector
return make_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNCDE/Make_model.py", line 16, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNCDE/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 44, in model_selector
return make_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNCDE/Make_model.py", line 16, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNCDE/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 44, in model_selector
return make_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNCDE/Make_model.py", line 16, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNCDE/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 44, in model_selector
return make_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNCDE/Make_model.py", line 16, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNCDE/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 44, in model_selector
return make_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNCDE/Make_model.py", line 16, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNCDE/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 44, in model_selector
return make_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNCDE/Make_model.py", line 16, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNCDE/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 44, in model_selector
return make_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNCDE/Make_model.py", line 16, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNCDE/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 44, in model_selector
return make_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNCDE/Make_model.py", line 16, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNCDE/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 44, in model_selector
return make_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNCDE/Make_model.py", line 16, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/NYCBike-InFlow.yaml
```
模型参数量: 103513539
加载 NYCBike-InFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/AEPSA/NYCBike-OutFlow.yaml
```
模型参数量: 103513539
加载 NYCBike-OutFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/ST_SSL/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 78, in model_selector
return STSSLModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ST_SSL/ST_SSL.py", line 39, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/ST_SSL/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 78, in model_selector
return STSSLModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ST_SSL/ST_SSL.py", line 39, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/ST_SSL/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 78, in model_selector
return STSSLModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ST_SSL/ST_SSL.py", line 39, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/ST_SSL/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 78, in model_selector
return STSSLModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ST_SSL/ST_SSL.py", line 39, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/ST_SSL/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 78, in model_selector
return STSSLModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ST_SSL/ST_SSL.py", line 39, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/ST_SSL/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 78, in model_selector
return STSSLModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ST_SSL/ST_SSL.py", line 39, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/ST_SSL/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 78, in model_selector
return STSSLModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ST_SSL/ST_SSL.py", line 39, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/ST_SSL/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 78, in model_selector
return STSSLModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ST_SSL/ST_SSL.py", line 39, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/ST_SSL/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 78, in model_selector
return STSSLModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ST_SSL/ST_SSL.py", line 39, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TCN/AirQuality.yaml
```
模型参数量: 36678
加载 AirQuality 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/AirQuality/2025-12-01_22-00-37/run.log
2025/12/01 22:00:37 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/TCN/AirQuality/2025-12-01_22-00-37
2025/12/01 22:00:37 - Training process started
Train Epoch 1: 0%| | 0/325 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/325 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TCN/TCN.py", line 43, in forward
x = self.network(x)
^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/container.py", line 250, in forward
input = module(input)
^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TCN/TCN.py", line 89, in forward
res = x if self.downsample is None else self.downsample(x)
^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 554, in forward
return self._conv_forward(input, self.weight, self.bias)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/conv.py", line 549, in _conv_forward
return F.conv2d(
^^^^^^^^^
RuntimeError: Given groups=1, weight of size [32, 6, 1, 1], expected input[16, 1, 35, 24] to have 6 channels, but got 1 channels instead
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 74, in model_selector
return EXP(model_config)
^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/EXP/EXP32.py", line 114, in __init__
self.horizon = args["horizon"] # 预测步长
~~~~^^^^^^^^^^^
KeyError: 'horizon'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 74, in model_selector
return EXP(model_config)
^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/EXP/EXP32.py", line 114, in __init__
self.horizon = args["horizon"] # 预测步长
~~~~^^^^^^^^^^^
KeyError: 'horizon'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 74, in model_selector
return EXP(model_config)
^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/EXP/EXP32.py", line 114, in __init__
self.horizon = args["horizon"] # 预测步长
~~~~^^^^^^^^^^^
KeyError: 'horizon'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 74, in model_selector
return EXP(model_config)
^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/EXP/EXP32.py", line 114, in __init__
self.horizon = args["horizon"] # 预测步长
~~~~^^^^^^^^^^^
KeyError: 'horizon'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/SD.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 74, in model_selector
return EXP(model_config)
^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/EXP/EXP32.py", line 114, in __init__
self.horizon = args["horizon"] # 预测步长
~~~~^^^^^^^^^^^
KeyError: 'horizon'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 74, in model_selector
return EXP(model_config)
^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/EXP/EXP32.py", line 114, in __init__
self.horizon = args["horizon"] # 预测步长
~~~~^^^^^^^^^^^
KeyError: 'horizon'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/PEMSD8.yaml
```
模型参数量: 235788
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 18, in get_dataloader
return EXP_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/EXPdataloader.py", line 8, in get_dataloader
data = load_st_dataset(args["type"], args["sample"]) # [T, N, F]
~~~~^^^^^^^^
KeyError: 'type'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 74, in model_selector
return EXP(model_config)
^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/EXP/EXP32.py", line 114, in __init__
self.horizon = args["horizon"] # 预测步长
~~~~^^^^^^^^^^^
KeyError: 'horizon'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 74, in model_selector
return EXP(model_config)
^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/EXP/EXP32.py", line 114, in __init__
self.horizon = args["horizon"] # 预测步长
~~~~^^^^^^^^^^^
KeyError: 'horizon'
```
#### /user/czzhangheng/code/TrafficWheel/config/EXP/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 74, in model_selector
return EXP(model_config)
^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/EXP/EXP32.py", line 114, in __init__
self.horizon = args["horizon"] # 预测步长
~~~~^^^^^^^^^^^
KeyError: 'horizon'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD7(L).yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/Hainan.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DDGCRN/PEMSD7(M).yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 36, in model_selector
return DDGCRN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/DDGCRN/DDGCRN.py", line 41, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/DSANET/NYCBike-InFlow.yaml
```
模型参数量: 37897240
加载 NYCBike-InFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/DSANET/AirQuality.yaml
```
模型参数量: 37897240
加载 AirQuality 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/DSANET/AirQuality/2025-12-01_22-05-16/run.log
2025/12/01 22:05:16 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/DSANET/AirQuality/2025-12-01_22-05-16
2025/12/01 22:05:16 - Training process started
Train Epoch 1: 0%| | 0/325 [00:00<?, ?it/s]/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/loss.py:128: UserWarning: Using a target size (torch.Size([16, 24, 35, 6])) that is different to the input size (torch.Size([16, 12, 35, 1])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.
return F.l1_loss(input, target, reduction=self.reduction)
Train Epoch 1: 0%| | 0/325 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 180, in _run_epoch
loss = self.loss(output, label)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/loss.py", line 128, in forward
return F.l1_loss(input, target, reduction=self.reduction)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/functional.py", line 3810, in l1_loss
expanded_input, expanded_target = torch.broadcast_tensors(input, target)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/functional.py", line 76, in broadcast_tensors
return _VF.broadcast_tensors(tensors) # type: ignore[attr-defined]
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: The size of tensor a (12) must match the size of tensor b (24) at non-singleton dimension 1
```
#### /user/czzhangheng/code/TrafficWheel/config/DSANET/NYCBike-OutFlow.yaml
```
模型参数量: 37897240
加载 NYCBike-OutFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/DSANET/SolarEnergy.yaml
```
模型参数量: 37897240
加载 SolarEnergy 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/DSANET/SolarEnergy/2025-12-01_22-06-16/run.log
2025/12/01 22:06:16 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/DSANET/SolarEnergy/2025-12-01_22-06-16
2025/12/01 22:06:16 - Training process started
Train Epoch 1: 0%| | 0/1970 [00:00<?, ?it/s]/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/loss.py:128: UserWarning: Using a target size (torch.Size([16, 24, 137, 3])) that is different to the input size (torch.Size([16, 12, 137, 1])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.
return F.l1_loss(input, target, reduction=self.reduction)
Train Epoch 1: 0%| | 0/1970 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 180, in _run_epoch
loss = self.loss(output, label)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/loss.py", line 128, in forward
return F.l1_loss(input, target, reduction=self.reduction)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/functional.py", line 3810, in l1_loss
expanded_input, expanded_target = torch.broadcast_tensors(input, target)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/functional.py", line 76, in broadcast_tensors
return _VF.broadcast_tensors(tensors) # type: ignore[attr-defined]
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: The size of tensor a (12) must match the size of tensor b (24) at non-singleton dimension 1
```
#### /user/czzhangheng/code/TrafficWheel/config/STFGNN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 58, in model_selector
return STFGNN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STFGNN/STFGNN.py", line 343, in __init__
adj = torch.tensor(get_adj(args))
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STFGNN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 58, in model_selector
return STFGNN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STFGNN/STFGNN.py", line 343, in __init__
adj = torch.tensor(get_adj(args))
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STFGNN/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 58, in model_selector
return STFGNN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STFGNN/STFGNN.py", line 343, in __init__
adj = torch.tensor(get_adj(args))
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STFGNN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 58, in model_selector
return STFGNN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STFGNN/STFGNN.py", line 343, in __init__
adj = torch.tensor(get_adj(args))
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STFGNN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 58, in model_selector
return STFGNN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STFGNN/STFGNN.py", line 343, in __init__
adj = torch.tensor(get_adj(args))
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STFGNN/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 58, in model_selector
return STFGNN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STFGNN/STFGNN.py", line 343, in __init__
adj = torch.tensor(get_adj(args))
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STFGNN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 58, in model_selector
return STFGNN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STFGNN/STFGNN.py", line 343, in __init__
adj = torch.tensor(get_adj(args))
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STFGNN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 58, in model_selector
return STFGNN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STFGNN/STFGNN.py", line 343, in __init__
adj = torch.tensor(get_adj(args))
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STFGNN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 58, in model_selector
return STFGNN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STFGNN/STFGNN.py", line 343, in __init__
adj = torch.tensor(get_adj(args))
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/AGCRN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 40, in model_selector
return AGCRN(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/AGCRN/AGCRN.py", line 52, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/AGCRN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 40, in model_selector
return AGCRN(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/AGCRN/AGCRN.py", line 52, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/AGCRN/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 40, in model_selector
return AGCRN(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/AGCRN/AGCRN.py", line 52, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/AGCRN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 40, in model_selector
return AGCRN(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/AGCRN/AGCRN.py", line 52, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/AGCRN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 40, in model_selector
return AGCRN(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/AGCRN/AGCRN.py", line 52, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/AGCRN/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 40, in model_selector
return AGCRN(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/AGCRN/AGCRN.py", line 52, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/AGCRN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 40, in model_selector
return AGCRN(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/AGCRN/AGCRN.py", line 52, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/AGCRN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 40, in model_selector
return AGCRN(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/AGCRN/AGCRN.py", line 53, in __init__
self.input_dim = args["input_dim"]
~~~~^^^^^^^^^^^^^
KeyError: 'input_dim'
```
#### /user/czzhangheng/code/TrafficWheel/config/AGCRN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 40, in model_selector
return AGCRN(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/AGCRN/AGCRN.py", line 52, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNRDE/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 80, in model_selector
return make_nrde_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNRDE/Make_model.py", line 17, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNRDE/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 80, in model_selector
return make_nrde_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNRDE/Make_model.py", line 17, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNRDE/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 80, in model_selector
return make_nrde_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNRDE/Make_model.py", line 17, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNRDE/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 80, in model_selector
return make_nrde_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNRDE/Make_model.py", line 17, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNRDE/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 80, in model_selector
return make_nrde_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNRDE/Make_model.py", line 17, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNRDE/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 80, in model_selector
return make_nrde_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNRDE/Make_model.py", line 17, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNRDE/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 80, in model_selector
return make_nrde_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNRDE/Make_model.py", line 17, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNRDE/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 80, in model_selector
return make_nrde_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNRDE/Make_model.py", line 17, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGNRDE/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 80, in model_selector
return make_nrde_model(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGNRDE/Make_model.py", line 17, in make_model
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/PEMS-BAY_paper.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 84, in model_selector
return REPST(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/REPST/repst.py", line 24, in __init__
self.word_choice = GumbelSoftmax(configs['word_num'])
~~~~~~~^^^^^^^^^^^^
KeyError: 'word_num'
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/NYCBike-InFlow.yaml
```
模型参数量: 103481647
加载 NYCBike-InFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/REPST/NYCBike-OutFlow.yaml
```
模型参数量: 103481647
加载 NYCBike-OutFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/STIDGCN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 68, in model_selector
return STIDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STIDGCN/STIDGCN.py", line 337, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STIDGCN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 68, in model_selector
return STIDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STIDGCN/STIDGCN.py", line 337, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STIDGCN/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 68, in model_selector
return STIDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STIDGCN/STIDGCN.py", line 337, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STIDGCN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 68, in model_selector
return STIDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STIDGCN/STIDGCN.py", line 337, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STIDGCN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 68, in model_selector
return STIDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STIDGCN/STIDGCN.py", line 337, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STIDGCN/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 68, in model_selector
return STIDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STIDGCN/STIDGCN.py", line 337, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STIDGCN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 68, in model_selector
return STIDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STIDGCN/STIDGCN.py", line 337, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STIDGCN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 68, in model_selector
return STIDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STIDGCN/STIDGCN.py", line 337, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STIDGCN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 68, in model_selector
return STIDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STIDGCN/STIDGCN.py", line 337, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 64, in model_selector
return PDG2Seq(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/PDG2SEQ/PDG2Seqb.py", line 226, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 64, in model_selector
return PDG2Seq(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/PDG2SEQ/PDG2Seqb.py", line 226, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 64, in model_selector
return PDG2Seq(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/PDG2SEQ/PDG2Seqb.py", line 226, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 64, in model_selector
return PDG2Seq(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/PDG2SEQ/PDG2Seqb.py", line 226, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 64, in model_selector
return PDG2Seq(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/PDG2SEQ/PDG2Seqb.py", line 226, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 64, in model_selector
return PDG2Seq(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/PDG2SEQ/PDG2Seqb.py", line 226, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 64, in model_selector
return PDG2Seq(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/PDG2SEQ/PDG2Seqb.py", line 226, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 64, in model_selector
return PDG2Seq(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/PDG2SEQ/PDG2Seqb.py", line 226, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/PDG2SEQ/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 64, in model_selector
return PDG2Seq(model_config)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/PDG2SEQ/PDG2Seqb.py", line 226, in __init__
self.num_nodes = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/NLT/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 42, in model_selector
return HierAttnLstm(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/NLT/HierAttnLstm.py", line 10, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/NLT/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 42, in model_selector
return HierAttnLstm(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/NLT/HierAttnLstm.py", line 10, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/NLT/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 42, in model_selector
return HierAttnLstm(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/NLT/HierAttnLstm.py", line 10, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/NLT/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 42, in model_selector
return HierAttnLstm(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/NLT/HierAttnLstm.py", line 10, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/NLT/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 42, in model_selector
return HierAttnLstm(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/NLT/HierAttnLstm.py", line 10, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/NLT/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 42, in model_selector
return HierAttnLstm(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/NLT/HierAttnLstm.py", line 10, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/NLT/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 42, in model_selector
return HierAttnLstm(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/NLT/HierAttnLstm.py", line 10, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/NLT/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 42, in model_selector
return HierAttnLstm(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/NLT/HierAttnLstm.py", line 10, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/NLT/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 42, in model_selector
return HierAttnLstm(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/NLT/HierAttnLstm.py", line 10, in __init__
args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/NYCBike-InFlow.yaml
```
模型参数量: 4
加载 NYCBike-InFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD4.yaml
```
模型参数量: 4
加载 PEMSD4 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/PEMSD4/2025-12-01_22-14-54/run.log
2025/12/01 22:14:54 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/PEMSD4/2025-12-01_22-14-54
2025/12/01 22:14:54 - Training process started
Train Epoch 1: 0%| | 0/160 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/160 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 40, in forward
drift = self.drift[n] if self.drift is not None else None
~~~~~~~~~~^^^
IndexError: index 1 is out of bounds for dimension 0 with size 1
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/METR-LA.yaml
```
模型参数量: 4
加载 METR-LA 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/METR-LA/2025-12-01_22-15-07/run.log
2025/12/01 22:15:07 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/METR-LA/2025-12-01_22-15-07
2025/12/01 22:15:07 - Training process started
Train Epoch 1: 0%| | 0/1285 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/1285 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 40, in forward
drift = self.drift[n] if self.drift is not None else None
~~~~~~~~~~^^^
IndexError: index 1 is out of bounds for dimension 0 with size 1
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/AirQuality.yaml
```
模型参数量: 4
加载 AirQuality 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/AirQuality/2025-12-01_22-15-15/run.log
2025/12/01 22:15:15 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/AirQuality/2025-12-01_22-15-15
2025/12/01 22:15:15 - Training process started
Train Epoch 1: 0%| | 0/325 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/325 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 40, in forward
drift = self.drift[n] if self.drift is not None else None
~~~~~~~~~~^^^
IndexError: index 1 is out of bounds for dimension 0 with size 1
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/NYCBike-OutFlow.yaml
```
模型参数量: 4
加载 NYCBike-OutFlow 数据集中...
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 22, in main
train_loader, val_loader, test_loader, scaler, *extra_data = get_dataloader(
^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/loader_selector.py", line 20, in get_dataloader
return normal_loader(config, normalizer, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 14, in get_dataloader
x, y = _prepare_data_with_windows(data, args, single)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 43, in _prepare_data_with_windows
x = _add_time_features(x, time_features, args["lag"], args["horizon"], single, add_window_x)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/dataloader/PeMSDdataloader.py", line 64, in _add_time_features
return np.concatenate([data, time_day, time_week], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 2, the array at index 0 has size 128 and the array at index 1 has size 1024
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD8.yaml
```
模型参数量: 4
加载 PEMSD8 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/PEMSD8/2025-12-01_22-15-31/run.log
2025/12/01 22:15:31 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/PEMSD8/2025-12-01_22-15-31
2025/12/01 22:15:31 - Training process started
Train Epoch 1: 0%| | 0/168 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/168 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 40, in forward
drift = self.drift[n] if self.drift is not None else None
~~~~~~~~~~^^^
IndexError: index 1 is out of bounds for dimension 0 with size 1
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD7(L).yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 52, in model_selector
return ARIMA(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 9, in __init__
self.p = args["p"] # 自回归阶数
~~~~^^^^^
KeyError: 'p'
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD3.yaml
```
模型参数量: 4
加载 PEMSD3 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/PEMSD3/2025-12-01_22-15-53/run.log
2025/12/01 22:15:53 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/PEMSD3/2025-12-01_22-15-53
2025/12/01 22:15:53 - Training process started
Train Epoch 1: 0%| | 0/982 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/982 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 40, in forward
drift = self.drift[n] if self.drift is not None else None
~~~~~~~~~~^^^
IndexError: index 1 is out of bounds for dimension 0 with size 1
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/SolarEnergy.yaml
```
模型参数量: 4
加载 SolarEnergy 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/SolarEnergy/2025-12-01_22-16-18/run.log
2025/12/01 22:16:18 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/SolarEnergy/2025-12-01_22-16-18
2025/12/01 22:16:18 - Training process started
Train Epoch 1: 0%| | 0/1970 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/1970 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 40, in forward
drift = self.drift[n] if self.drift is not None else None
~~~~~~~~~~^^^
IndexError: index 1 is out of bounds for dimension 0 with size 1
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/Hainan.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 52, in model_selector
return ARIMA(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 9, in __init__
self.p = args["p"] # 自回归阶数
~~~~^^^^^
KeyError: 'p'
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD7.yaml
```
模型参数量: 4
加载 PEMSD7 数据集中...
Create Log File in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/PEMSD7/2025-12-01_22-16-56/run.log
2025/12/01 22:16:56 - Experiment log path in: /user/czzhangheng/code/TrafficWheel/experiments/ARIMA/PEMSD7/2025-12-01_22-16-56
2025/12/01 22:16:56 - Training process started
Train Epoch 1: 0%| | 0/1058 [00:00<?, ?it/s]
Train Epoch 1: 0%| | 0/1058 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 46, in main
trainer.train()
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 266, in train
train_epoch_loss = self.train_epoch(epoch)
^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 244, in train_epoch
return self._run_epoch(epoch, self.train_loader, "train")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/trainer/Trainer.py", line 179, in _run_epoch
output = self.model(data).to(self.device)
^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/miniconda/envs/TrafficWheel/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 40, in forward
drift = self.drift[n] if self.drift is not None else None
~~~~~~~~~~^^^
IndexError: index 1 is out of bounds for dimension 0 with size 1
```
#### /user/czzhangheng/code/TrafficWheel/config/ARIMA/PEMSD7(M).yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 52, in model_selector
return ARIMA(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/ARIMA/ARIMA.py", line 9, in __init__
self.p = args["p"] # 自回归阶数
~~~~^^^^^
KeyError: 'p'
```
#### /user/czzhangheng/code/TrafficWheel/config/STMLP/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 66, in model_selector
return STMLP(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STMLP/STMLP.py", line 188, in __init__
self.adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STMLP/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 66, in model_selector
return STMLP(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STMLP/STMLP.py", line 188, in __init__
self.adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STMLP/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 66, in model_selector
return STMLP(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STMLP/STMLP.py", line 188, in __init__
self.adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STMLP/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 66, in model_selector
return STMLP(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STMLP/STMLP.py", line 188, in __init__
self.adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STMLP/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 66, in model_selector
return STMLP(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STMLP/STMLP.py", line 188, in __init__
self.adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STMLP/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 66, in model_selector
return STMLP(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STMLP/STMLP.py", line 188, in __init__
self.adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STMLP/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 66, in model_selector
return STMLP(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STMLP/STMLP.py", line 188, in __init__
self.adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STMLP/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 66, in model_selector
return STMLP(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STMLP/STMLP.py", line 188, in __init__
self.adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STMLP/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 66, in model_selector
return STMLP(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STMLP/STMLP.py", line 188, in __init__
self.adj_mx = get_adj(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 10, in get_adj
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/MegaCRN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 76, in model_selector
return MegaCRNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/MegaCRN/MegaCRNModel.py", line 32, in __init__
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/MegaCRN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 76, in model_selector
return MegaCRNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/MegaCRN/MegaCRNModel.py", line 32, in __init__
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/MegaCRN/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 76, in model_selector
return MegaCRNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/MegaCRN/MegaCRNModel.py", line 32, in __init__
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/MegaCRN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 76, in model_selector
return MegaCRNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/MegaCRN/MegaCRNModel.py", line 32, in __init__
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/MegaCRN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 76, in model_selector
return MegaCRNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/MegaCRN/MegaCRNModel.py", line 32, in __init__
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/MegaCRN/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 76, in model_selector
return MegaCRNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/MegaCRN/MegaCRNModel.py", line 32, in __init__
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/MegaCRN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 76, in model_selector
return MegaCRNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/MegaCRN/MegaCRNModel.py", line 32, in __init__
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/MegaCRN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 76, in model_selector
return MegaCRNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/MegaCRN/MegaCRNModel.py", line 32, in __init__
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/MegaCRN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 76, in model_selector
return MegaCRNModel(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/MegaCRN/MegaCRNModel.py", line 32, in __init__
num_nodes=args["num_nodes"],
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/GWN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 56, in model_selector
return gwnet(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/GWN/GraphWaveNet.py", line 91, in __init__
torch.randn(args["num_nodes"], 10, device=args["device"])
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/GWN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 56, in model_selector
return gwnet(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/GWN/GraphWaveNet.py", line 91, in __init__
torch.randn(args["num_nodes"], 10, device=args["device"])
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/GWN/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 56, in model_selector
return gwnet(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/GWN/GraphWaveNet.py", line 91, in __init__
torch.randn(args["num_nodes"], 10, device=args["device"])
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/GWN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 56, in model_selector
return gwnet(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/GWN/GraphWaveNet.py", line 91, in __init__
torch.randn(args["num_nodes"], 10, device=args["device"])
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/GWN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 56, in model_selector
return gwnet(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/GWN/GraphWaveNet.py", line 91, in __init__
torch.randn(args["num_nodes"], 10, device=args["device"])
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/GWN/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 56, in model_selector
return gwnet(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/GWN/GraphWaveNet.py", line 91, in __init__
torch.randn(args["num_nodes"], 10, device=args["device"])
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/GWN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 56, in model_selector
return gwnet(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/GWN/GraphWaveNet.py", line 91, in __init__
torch.randn(args["num_nodes"], 10, device=args["device"])
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/GWN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 56, in model_selector
return gwnet(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/GWN/GraphWaveNet.py", line 91, in __init__
torch.randn(args["num_nodes"], 10, device=args["device"])
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/GWN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 56, in model_selector
return gwnet(model_config)
^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/GWN/GraphWaveNet.py", line 91, in __init__
torch.randn(args["num_nodes"], 10, device=args["device"])
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGCN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 48, in model_selector
return STGCNChebGraphConv(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGCN/models.py", line 31, in __init__
gso = get_gso(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 33, in get_gso
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGCN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 48, in model_selector
return STGCNChebGraphConv(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGCN/models.py", line 31, in __init__
gso = get_gso(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 33, in get_gso
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGCN/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 48, in model_selector
return STGCNChebGraphConv(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGCN/models.py", line 31, in __init__
gso = get_gso(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 33, in get_gso
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGCN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 48, in model_selector
return STGCNChebGraphConv(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGCN/models.py", line 31, in __init__
gso = get_gso(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 33, in get_gso
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGCN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 48, in model_selector
return STGCNChebGraphConv(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGCN/models.py", line 31, in __init__
gso = get_gso(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 33, in get_gso
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGCN/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 48, in model_selector
return STGCNChebGraphConv(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGCN/models.py", line 31, in __init__
gso = get_gso(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 33, in get_gso
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGCN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 48, in model_selector
return STGCNChebGraphConv(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGCN/models.py", line 31, in __init__
gso = get_gso(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 33, in get_gso
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGCN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 48, in model_selector
return STGCNChebGraphConv(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGCN/models.py", line 31, in __init__
gso = get_gso(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 33, in get_gso
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STGCN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 48, in model_selector
return STGCNChebGraphConv(model_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/STGCN/models.py", line 31, in __init__
gso = get_gso(args)
^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/get_adj.py", line 33, in get_gso
match args['num_nodes']:
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```