[Refactor]: Change network_blocks to trasformer_blocks

open-mmlab · Apr 22, 2022 · 9cd0ae2 · 9cd0ae2
1 parent 5270ae8
commit 9cd0ae2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mmselfsup/models/utils/__init__.py b/mmselfsup/models/utils/__init__.py
@@ -6,8 +6,8 @@
 from .knn_classifier import knn_classifier
 from .multi_pooling import MultiPooling
 from .multi_prototypes import MultiPrototypes
-from .network_blocks import (CAETransformerRegressorLayer, MultiheadAttention,
-                             TransformerEncoderLayer)
+from .transformer_blocks import (CAETransformerRegressorLayer,
+                                 MultiheadAttention, TransformerEncoderLayer)
 from .position_embedding import build_2d_sincos_position_embedding
 from .sobel import Sobel
 

diff --git a/mmselfsup/models/utils/network_blocks.py → mmselfsup/models/utils/transformer_blocks.py b/mmselfsup/models/utils/network_blocks.py → mmselfsup/models/utils/transformer_blocks.py