# 测试报告 ## 测试概述 - 测试时间: 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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' ```