Fix README.md
This commit is contained in:
parent
2fed8f3b37
commit
5127ba2a27
1 changed files with 1 additions and 0 deletions
|
@ -9,6 +9,7 @@ data = dataset[0]
|
|||
|
||||
import torch.nn.functional as F
|
||||
from torch_geometric.nn import GCNConv, global_mean_pool
|
||||
from eixgnn.eixgnn import EiXGNN
|
||||
|
||||
class GCN(torch.nn.Module):
|
||||
def __init__(self):
|
||||
|
|
Loading…
Add table
Reference in a new issue