class_docstr: Impala Network which also outputs penultimate representation layers.
class_docstr_tok: ["Impala","Network","which","also","outputs","penultimate","representation","layers","."]
code: def setup(self):
      self.encoder = ImpalaEncoder(nn_scale=self.nn_scale)
  
imports: import collections
  from typing import Tuple
  from absl import logging
  from dopamine.discrete_domains import atari_lib
  from flax import linen
  import gin
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling the ImpalaNetworkWithRepresentations obj's underlying member method code with example usage: obj.setup() without return types