Skip to content

Commit b3cba5e

Browse files
committed
update
1 parent b5d3c89 commit b3cba5e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch_geometric/io/planetoid.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def read_file(folder, prefix, name):
103103
return out
104104

105105
out = out.todense() if hasattr(out, 'todense') else out
106-
out = torch.Tensor(out)
106+
out = torch.from_numpy(out).to(torch.float)
107107
return out
108108

109109

0 commit comments

Comments
 (0)