PoliBa-DeepLearning/NanoSocrates
2
0
Fork 0
Code Issues 3 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Changed Imports

Browse Source
This commit is contained in:
Christian Risi 2025-10-07 16:36:59 +02:00
parent fef933da9d
commit 109ad9f36b
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Project_Model/Libs/Transformer/Classes/__init__.py
View File

@ -1,15 +1,16 @@
from .Decoder import Decoder from .Decoder import Decoder
from .Encoder import Encoder from .Encoder import Encoder
from .FeedForwardNetwork import FeedForwardNetwork from .FeedForwardNetwork import FeedForwardNetwork
from .MultiHeadAttention import MultiheadAttention # from .MultiHeadAttention import MultiheadAttention
from .TorchMultiHeadAttention import TorchMultiHeadAttention from .TorchMultiHeadAttention import TorchMultiHeadAttention
from .SpannedMasker import SpannedMasker from .SpannedMasker import SpannedMasker
from .DeToken import DeToken
__all__ = [ __all__ = [
"Decoder", "Decoder",
"Encoder", "Encoder",
"FeedForwardNetwork", "FeedForwardNetwork",
"MultiheadAttention",
"TorchMultiHeadAttention", "TorchMultiHeadAttention",
"SpannedMasker" "SpannedMasker",
"DeToken"
] ]