Browse Source

fix bug in pyg gin base.py

tags/v0.3.1
SwiftieH 4 years ago
parent
commit
8e03d18dbe
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      test/performance/graph_classification/pyg/base.py

+ 1
- 1
test/performance/graph_classification/pyg/base.py View File

@@ -10,7 +10,7 @@ import torch.nn.functional as F
from torch.nn import Sequential, Linear, ReLU
import torch_geometric
from torch_geometric.datasets import TUDataset
if int(torch_geometric.__version__.split(",")[0]) >= 2:
if int(torch_geometric.__version__.split(".")[0]) >= 2:
from torch_geometric.loader import DataLoader
else:
from torch_geometric.data import DataLoader


Loading…
Cancel
Save