code: def __init__(self, act_dim):
      self.conv1 = layers.conv2d(
          num_filters=16, filter_size=4, stride=2, padding=1, act="relu"
      )
      self.conv2 = layers.conv2d(
          num_filters=32, filter_size=4, stride=2, padding=2, act="relu"
      )
      self.conv3 = layers.conv2d(
          num_filters=256,
          filter_size=11,
          stride=1,
          padding=0,
          act="relu",
      )
  
      self.policy_conv = layers.conv2d(
          num_filters=act_dim,
          filter_size=1,
          stride=1,
          padding=0,
          act=None,
          param_attr=ParamAttr(initializer=fluid.initializer.Normal()),
      )
  
      self.value_fc = layers.fc(
          size=1,
          param_attr=ParamAttr(initializer=fluid.initializer.Normal()),
      )
  
file_path: PARL/benchmark/fluid/IMPALA/atari_model.py
imports: import paddle.fluid
  import parl
  from parl import layers
  from paddle.fluid.param_attr import ParamAttr
label_desc: Use image node_id 1 to create a new AtariModel object from inherited base classes: parl with example: obj = AtariModel(act_dim)