duycse1603's picture
[Add] source
6163604
from .seq2seq import Attention
from .seq2seq_v2 import AttentionV2
from .tfm import TransformerPrediction
__all__ = ['Attention', 'AttentionV2', 'TransformerPrediction']