code: def __init__(self, num_actions=2):
      super(Model, self).__init__()
  
      self.linear0 = nn.Linear(4, 128)
      self.linear1 = nn.Linear(128, 32)
  
      core_output_size = self.linear1.out_features
  
      if USE_LSTM:
          self.core = nn.LSTMCell(core_output_size, 32)
          core_output_size = 32
  
      self.policy = nn.Linear(32, num_actions)
      self.baseline = nn.Linear(32, 1)
  
file_path: moolib/examples/a2c.py
imports: import csv
  import os
  import time
  import gym
  import torch
  from torch import nn
  from torch.nn import functional
  import moolib
label_desc: Use image node_id 1 to create a new Model object from inherited base classes: nn with example: obj = Model(num_actions)