Skip to content

A
AIIM2023-GNNs
Commit 3a7c6c2d authored by ADRIAN AYUSO MUNOZ's avatar ADRIAN AYUSO MUNOZ
Browse files
Options

Name change

parent 448ca4e4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
topN.py
View file @ 3a7c6c2d
......@@ -65,7 +65,7 @@ def deepSnap():
convs = generate_convs_link_pred_layers(hetero, HeteroSAGEConv, hidden_dim)
model = HeteroGNN(convs, hetero, hidden_dim, dropout).to(device).to(device)
model.load_state_dict(torch.load("./models/modelDeepSnapPredFull", map_location=torch.device(device)))
model.load_state_dict(torch.load("./models/behor", map_location=torch.device(device)))
model = model.to(device)
model.eval()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment