code: def __init__(self, obs_space, act_space):
      super(AtariModel, self).__init__()
  
      self.conv1 = _init_layer(nn.Conv2d(4, 32, 8, stride=4))
      self.conv2 = _init_layer(nn.Conv2d(32, 64, 4, stride=2))
      self.conv3 = _init_layer(nn.Conv2d(64, 64, 3, stride=1))
  
      self.relu = nn.ReLU()
      self.flatten = nn.Flatten()
      self.fc = _init_layer(nn.Linear(64 * 7 * 7, 512))
  
      self.fc_pi = _init_layer(nn.Linear(512, act_space.n), std=0.01)
      self.fc_v = _init_layer(nn.Linear(512, 1), std=1)
  
file_path: PARL/benchmark/torch/ppo/atari_model.py
imports: import parl
  import torch
  import torch.nn
  import numpy
label_desc: Use image node_id 1 to create a new AtariModel object from inherited base classes: parl with example: obj = AtariModel(obs_space, act_space)