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

修复模型 配置错误

This commit is contained in:
czzhangheng 2025-11-27 19:21:01 +08:00
parent 35310a4f4a
commit 0ad1494aec
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/AEPSA/NYCBike-inflow.yaml
View File

@ -1,5 +1,5 @@
basic: basic:
dataset: "NYCBike-inflow" dataset: "NYCBike-InFlow"
mode : "train" mode : "train"
device : "cuda:0" device : "cuda:0"
model: "AEPSA" model: "AEPSA"

2
config/AEPSA/NYCBike-outflow.yaml
View File

@ -1,5 +1,5 @@
basic: basic:
dataset: "NYCBike-outflow" dataset: "NYCBike-OutFlow"
mode : "train" mode : "train"
device : "cuda:0" device : "cuda:0"
model: "AEPSA" model: "AEPSA"