code: def __init__(self, game, args):
      # game params
      self.board_x, self.board_y = game.getBoardSize()
      self.action_size = game.getActionSize()
      self.args = args
  
      super(Connect4Model, self).__init__()
      self.conv1 = nn.Conv2D(
          1, args.num_channels, 3, stride=1, padding=1
      )
      self.conv2 = nn.Conv2D(
          args.num_channels, args.num_channels, 3, stride=1, padding=1
      )
      self.conv3 = nn.Conv2D(
          args.num_channels, args.num_channels, 3, stride=1
      )
      self.conv4 = nn.Conv2D(
          args.num_channels, args.num_channels, 3, stride=1
      )
  
      self.bn1 = nn.BatchNorm2D(args.num_channels)
      self.bn2 = nn.BatchNorm2D(args.num_channels)
      self.bn3 = nn.BatchNorm2D(args.num_channels)
      self.bn4 = nn.BatchNorm2D(args.num_channels)
  
      self.fc1 = nn.Linear(
          args.num_channels * (self.board_x - 4) * (self.board_y - 4),
          128,
      )
      self.fc_bn1 = nn.BatchNorm1D(128)
  
      self.fc2 = nn.Linear(128, 64)
      self.fc_bn2 = nn.BatchNorm1D(64)
  
      self.fc3 = nn.Linear(64, self.action_size)
  
      self.fc4 = nn.Linear(64, 1)
      self.tanh = nn.Tanh()
  
file_path: PARL/examples/AlphaZero/connect4_model.py
imports: import parl
  import paddle
  import paddle.nn
  import paddle.nn.functional
label_desc: Use image node_id 1 to create a new Connect4Model object from inherited base classes: parl with example: obj = Connect4Model(game, args)