Implementation of DataLoader and DataModule (#383)
Refactoring for 0.2 * Data module, data loader and dataset * Refactor LabelTensor * Refactor solvers Co-authored-by: dario-coscia <dariocos99@gmail.com>
This commit is contained in:
committed by
Nicola Demo
parent
dd43c8304c
commit
a27bd35443
@@ -93,8 +93,8 @@ class Graph:
|
||||
|
||||
logging.debug(f"edge_index computed")
|
||||
return Data(
|
||||
x=nodes_data,
|
||||
pos=nodes_coordinates,
|
||||
x=nodes_data.tensor,
|
||||
pos=nodes_coordinates.tensor,
|
||||
edge_index=edge_index,
|
||||
edge_attr=edges_data,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user