Exemple #1
0
    def call(self, x):
        residual = x

        x = forward_layer_list(x, self.body)

        if self.downsample is not None:
            residual = forward_layer_list(residual, self.downsample)

        return x + residual
Exemple #2
0
    def call(self, x):
        residual = x

        x = forward_layer_list(x, self.body)
        if self.downsample is not None:
            residual = forward_layer_list(residual, self.downsample)

        x = residual + x

        return tf.keras.activations.relu(x)
Exemple #3
0
    def call(self, x):
        x = forward_layer_list(x, self.features)
        x = self.output_layer(x)

        return x