text: #!/usr/bin/env python3
  import argparse
  import torch
  import torch.nn.functional as F
  from torch.nn import Linear
  from datasets import get_planetoid_dataset
  import pdb
  from torch_geometric.nn import GCNConv
  
  
  hidden=128
  class Net(torch.nn.Module):
      def __init__(self, dataset):
          super(Net, self).__init__()
          self.conv1 = GCNConv(dataset.num_features, hidden)
          self.conv2 = GCNConv(hidden, dataset.num_classes)
      def reset_parameters(self):
          self.conv1.reset_parameters()
          self.conv2.reset_parameters()
  
      def forward(self, data, edge_index):
          x= data.x
          x = F.relu(self.conv1(x, edge_index))
          x = self.conv2(x, edge_index)
          return  F.log_softmax(x, dim=1)
  
  