Fix README.md

This commit is contained in:
araison 2023-03-08 19:04:15 +01:00
parent 2fed8f3b37
commit 5127ba2a27
1 changed files with 1 additions and 0 deletions

View File

@ -9,6 +9,7 @@ data = dataset[0]
import torch.nn.functional as F import torch.nn.functional as F
from torch_geometric.nn import GCNConv, global_mean_pool from torch_geometric.nn import GCNConv, global_mean_pool
from eixgnn.eixgnn import EiXGNN
class GCN(torch.nn.Module): class GCN(torch.nn.Module):
def __init__(self): def __init__(self):