Update traffic_dataloader_v2.py

2024-11-28 10:13:15 +08:00 · 2024-11-28 10:13:15 +08:00 · 50785ad3c1
parent b9ca3349c9
commit 50785ad3c1
1 changed files with 0 additions and 1 deletions
--- a/federatedscope/trafficflow/dataloader/traffic_dataloader_v2.py
+++ b/federatedscope/trafficflow/dataloader/traffic_dataloader_v2.py
@ -89,7 +89,6 @@ def load_traffic_data(config, client_cfgs):
    root = config.data.root
    dataName = 'PEMSD' + root[-1]
    raw_data = load_st_dataset(dataName)
-    sub_graph_size = config.model.minigraph_size

    l, n, f = raw_data.shape