Commit 64203b12 authored by Carlos GO's avatar Carlos GO
Browse files

send k to device

parent 464dc769
......@@ -173,6 +173,7 @@ def train_model(model, criterion, optimizer, device, train_loader, test_loader,
if model.clustered:
fp = fp.long()
fp = fp.to(device)
K = K.to(device)
# for f, i in zip(fp, idx):
# print(all_graphs[i.item()], f)
......
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