Add: Add initialization operation of modules.

chairc · May 6, 2024 · 6e46b54 · 6e46b54
1 parent 52d6c12
commit 6e46b54
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/model/modules/__init__.py b/model/modules/__init__.py
@@ -5,3 +5,6 @@
     @Author : chairc
     @Site   : https://github.com/chairc
 """
+from .activation import get_activation_function
+from .attention import SelfAttention
+from .block import DownBlock, UpBlock, CSPDarkDownBlock, CSPDarkUpBlock