Update logic2.py
Browse files
logic2.py
CHANGED
|
@@ -66,6 +66,7 @@ def load_hetero_data():
|
|
| 66 |
return data
|
| 67 |
|
| 68 |
def load_model(data):
|
|
|
|
| 69 |
model = Model(hidden_channels=32)
|
| 70 |
with torch.no_grad():
|
| 71 |
model.encoder(data.x_dict, data.edge_index_dict)
|
|
|
|
| 66 |
return data
|
| 67 |
|
| 68 |
def load_model(data):
|
| 69 |
+
global data
|
| 70 |
model = Model(hidden_channels=32)
|
| 71 |
with torch.no_grad():
|
| 72 |
model.encoder(data.x_dict, data.edge_index_dict)
|