class_docstr: The convolutional network used to compute the agent's Q-values.
code: def setup(self):
      self.encoder = ImpalaEncoder(nn_scale=self.nn_scale)
  
imports: import itertools
  import time
  from typing import Optional, Sequence, Tuple, Union
  from absl import logging
  from dopamine.discrete_domains import atari_lib
  from flax import linen
  import gin
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling the ImpalaDQNNetwork obj's underlying member method code with example usage: obj.setup() without return types