import tensorflow as tf

        return X

    def _count_model_parameters(self):
        tf_trainable_vars = tf.trainable_variables()
        num_params = 0
        # utils.logger.log('Model parameters:')
