class_docstr: Adam optimizer
docstr: prepare optimizer for training
code: def setup(self, policy: TorchPolicy) -> None:
      """prepare optimizer for training"""
      super().setup(policy)
  
      self.m = np.zeros(self.dim, dtype=np.float32)
      self.v = np.zeros(self.dim, dtype=np.float32)
  
file_path: maze/maze/train/trainers/es/optimizers/adam.py
imports: from typing import Optional
  import numpy
  from maze.core.agent.torch_policy import TorchPolicy
  from maze.train.trainers.es.optimizers.base_optimizer import Optimizer
label_desc: Use image node_id 2 for calling the Adam obj's underlying member method code with example usage: obj.setup(policy) without return types