iicd/Graph-DiT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

make data to use 1 rather than 2 size list

Browse Source
This commit is contained in:
mhz 2024-09-03 22:20:13 +02:00
parent 8bbadce19c
commit dcfefb91a3
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
graph_dit/datasets/dataset.py
View File

@ -716,7 +716,6 @@ class DataInfos(AbstractDatasetInfos):
graphs.append((adj_matrix, ops)) graphs.append((adj_matrix, ops))
meta_dict = graphs_to_json(graphs, 'nasbench-201') meta_dict = graphs_to_json(graphs, 'nasbench-201')
self.base_path = base_path self.base_path = base_path
self.active_atoms = meta_dict['active_atoms'] self.active_atoms = meta_dict['active_atoms']
self.max_n_nodes = meta_dict['max_node'] self.max_n_nodes = meta_dict['max_node']