code: def policy(self, obs):
      obs = obs / 255.0
      out = self.relu(self.conv1(obs))
      out = self.relu(self.conv2(out))
      out = self.relu(self.conv3(out))
  
      out = self.flatten(out)
      out = self.relu(self.fc(out))
      logits = self.fc_pi(out)
      return logits
  
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 3 for calling the AtariModel obj's underlying member method code with example usage: obj.policy(obs) and returns: logits