finalize context rcnn tf2

b4aa41f5 · Kaushik Shivakumar · eef0a1e7 · b4aa41f5
Commit b4aa41f5 authored Jul 23, 2020 by Kaushik Shivakumar
Show whitespace changes
Inline Side-by-side

Showing with 10 additions and 9 deletions

research/object_detection/meta_architectures/context_rcnn_lib_tf2.py ...ject_detection/meta_architectures/context_rcnn_lib_tf2.py +10 -9

No files found.
--- a/research/object_detection/meta_architectures/context_rcnn_lib_tf2.py
+++ b/research/object_detection/meta_architectures/context_rcnn_lib_tf2.py
@@ -59,6 +59,7 @@ class AttentionBlock(tf.keras.layers.Layer):
        output feature.
      is_training: A boolean Tensor (affecting batch normalization).
      name: A string describing what to name the variables in this block.
+      **kwargs: Additional keyword arguments.
    """

    self._key_proj = ContextProjection(bottleneck_dimension)