code: def _build_module(self, input_layer):
      weight_init = Orthogonal(gain=np.sqrt(2))
      input_layer = Conv2d(num_filters=32, kernel_size=8, strides=4)(
          input_layer, kernel_initializer=weight_init
      )
      input_layer = BatchnormActivationDropout(
          activation_function=self.activation_function
      )(input_layer)[-1]
      input_layer = Conv2d(num_filters=64, kernel_size=4, strides=2)(
          input_layer, kernel_initializer=weight_init
      )
      input_layer = BatchnormActivationDropout(
          activation_function=self.activation_function
      )(input_layer)[-1]
      input_layer = Conv2d(num_filters=64, kernel_size=3, strides=1)(
          input_layer, kernel_initializer=weight_init
      )
      input_layer = BatchnormActivationDropout(
          activation_function=self.activation_function
      )(input_layer)[-1]
      input_layer = tf.contrib.layers.flatten(input_layer)
  
      if self.is_predictor:
          input_layer = self.dense_layer(512)(
              input_layer, kernel_initializer=weight_init
          )
          input_layer = BatchnormActivationDropout(
              activation_function=tf.nn.relu
          )(input_layer)[-1]
          input_layer = self.dense_layer(512)(
              input_layer, kernel_initializer=weight_init
          )
          input_layer = BatchnormActivationDropout(
              activation_function=tf.nn.relu
          )(input_layer)[-1]
  
      self.output = self.dense_layer(512)(
          input_layer, name="output", kernel_initializer=weight_init
      )
  
file_path: coach/rl_coach/architectures/tensorflow_components/heads/RND_head.py
imports: import tensorflow
  import numpy
  from rl_coach.architectures.tensorflow_components.layers import Conv2d, BatchnormActivationDropout
  from rl_coach.architectures.tensorflow_components.heads.head import Head, Orthogonal
  from rl_coach.base_parameters import AgentParameters
  from rl_coach.core_types import Embedding
  from rl_coach.spaces import SpacesDefinition
label_desc: Use image node_id 2 for calling the RNDHead obj's underlying member method code with example usage: obj._build_module(input_layer) without return types