Commit 6d39f5e6 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

finalize context RCNN changes for tensorflow 2.x

parent afd5579f
......@@ -85,9 +85,7 @@ class AttentionBlock(tf.keras.layers.Layer):
self._val_proj, normalize=True)
weights = tf.matmul(queries, keys, transpose_b=True)
weights, values = filter_weight_value(weights, values, valid_mask)
weights = tf.nn.softmax(weights / self._attention_temperature)
features = tf.matmul(weights, values)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment