Refactoring
This commit is contained in:
13
Project_Model/Libs/Transformer/Classes/__init__.py
Normal file
13
Project_Model/Libs/Transformer/Classes/__init__.py
Normal file
@@ -0,0 +1,13 @@
|
||||
from .Decoder import Decoder
|
||||
from .Encoder import Encoder
|
||||
from .FeedForwardNetwork import FeedForwardNetwork
|
||||
from .MultiHeadAttention import MultiheadAttention
|
||||
from .TorchMultiHeadAttention import TorchMultiHeadAttention
|
||||
|
||||
__all__ = [
|
||||
"Decoder",
|
||||
"Encoder",
|
||||
"FeedForwardNetwork",
|
||||
"MultiheadAttention",
|
||||
"TorchMultiHeadAttention"
|
||||
]
|
||||
Reference in New Issue
Block a user