code: def __init__(self, act_dim):
      super(ActorCritic, self).__init__()
      self.conv1 = nn.Conv2d(
          in_channels=4,
          out_channels=32,
          kernel_size=8,
          stride=4,
          padding=1,
      )
      self.conv2 = nn.Conv2d(
          in_channels=32,
          out_channels=64,
          kernel_size=4,
          stride=2,
          padding=2,
      )
      self.conv3 = nn.Conv2d(
          in_channels=64,
          out_channels=64,
          kernel_size=3,
          stride=1,
          padding=0,
      )
  
      self.flatten = nn.Flatten()
      self.fc = nn.Linear(64 * 9 * 9, 512)
  
      self.fc_pi = nn.Linear(512, act_dim)
      self.fc_v = nn.Linear(512, 1)
  
      # params init
      self._init_parameters()
  
file_path: PARL/benchmark/torch/a2c/atari_model.py
imports: import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  import parl
label_desc: Use image node_id 1 to create a new ActorCritic object from inherited base classes: parl with example: obj = ActorCritic(act_dim)