Merge 84c42d9 into 933b052

open-mmlab · Mar 24, 2021 · bc869e4 · bc869e4
2 parents 933b052 + 84c42d9
commit bc869e4
Show file tree

Hide file tree

Showing 2 changed files with 479 additions and 0 deletions.
diff --git a/mmcv/cnn/bricks/registry.py b/mmcv/cnn/bricks/registry.py
@@ -6,3 +6,8 @@
 PADDING_LAYERS = Registry('padding layer')
 UPSAMPLE_LAYERS = Registry('upsample layer')
 PLUGIN_LAYERS = Registry('plugin layer')
+
+POSITIONAL_ENCODING = Registry('Position encoding')
+ATTENTION = Registry('Attention')
+TRANSFORMER_LAYER = Registry('TransformerLayer')
+TRANSFORMER_LAYER_SEQUENCE = Registry('TransformerLayerSequence')