import tensorflow as tf

        x = tf.keras.layers.AveragePooling2D(pool_size=7, strides=1,
                                        padding="valid", name="pool")(x)

        x = tf.reshape(x, shape=(-1, 1024))

        self.logits = self.__fully_connected(name="fc_nsfw",
