code: def __init__(self, algorithm, config):
      super(Agent, self).__init__(algorithm)
      self.obs_shape = config["obs_shape"]
      self.lr_scheduler = LinearDecayScheduler(
          config["start_lr"], config["max_sample_steps"]
      )
  
      self.entropy_coeff_scheduler = PiecewiseScheduler(
          config["entropy_coeff_scheduler"]
      )
  
      self.device = torch.device(
          "cuda" if torch.cuda.is_available() else "cpu"
      )
  
file_path: PARL/benchmark/torch/a2c/atari_agent.py
imports: import torch
  import parl
  from parl.utils.scheduler import PiecewiseScheduler, LinearDecayScheduler
  import numpy
label_desc: Use image node_id 1 to create a new Agent object from inherited base classes: parl with example: obj = Agent(algorithm, config)