def _build_endpoints(self, net):
    'Match filter size for endpoints before sharing conv layers.'
    endpoints = {}
    for level in range(self._min_level, (self._max_level + 1)):
        x = layers.Conv2D(filters=self._endpoints_num_filters, kernel_size=1, strides=1, use_bias=False, kernel_initializer=self._kernel_initializer, kernel_regularizer=self._kernel_regularizer, bias_regularizer=self._bias_regularizer)(net[str(level)])
        x = self._norm(axis=self._bn_axis, momentum=self._norm_momentum, epsilon=self._norm_epsilon)(x)
        x = tf_utils.get_activation(self._activation_fn)(x)
        endpoints[str(level)] = x
    return endpoints
Match filter size for endpoints before sharing conv layers.
def _build_endpoints(self, net):
    
    endpoints = {}
    for level in range(self._min_level, (self._max_level + 1)):
        x = layers.Conv2D(filters=self._endpoints_num_filters, kernel_size=1, strides=1, use_bias=False, kernel_initializer=self._kernel_initializer, kernel_regularizer=self._kernel_regularizer, bias_regularizer=self._bias_regularizer)(net[str(level)])
        x = self._norm(axis=self._bn_axis, momentum=self._norm_momentum, epsilon=self._norm_epsilon)(x)
        x = tf_utils.get_activation(self._activation_fn)(x)
        endpoints[str(level)] = x
    return endpoints