def create_model(cls, en_config, de_config):
from blocks.decoders import TransformerDecoder
encoder = nn.Embedding(en_config['vocab_size'], en_config['d_model'])
decoder = TransformerDecoder(de_config)
model = cls(encoder, decoder)
model.encoder.config = en_config
return model
def create_model(cls, sp_config, en_config, de_config):
from blocks.sp_layers import SPLayer
from blocks.encoders import TransformerEncoder
from blocks.decoders import TransformerDecoder
splayer = SPLayer(sp_config)
encoder = TransformerEncoder(en_config)
decoder = TransformerDecoder(de_config)
model = cls(splayer, encoder, decoder)
return model
def create_model(cls, sp_config, en_config, as_cofig, phone_size, de_config):
from blocks.sp_layers import SPLayer
from blocks.encoders import TransformerEncoder
from blocks.attention_assigner import Attention_Assigner
from blocks.decoders import TransformerDecoder
splayer = SPLayer(sp_config)
encoder = TransformerEncoder(en_config)
assigner = Attention_Assigner(as_cofig)
decoder = TransformerDecoder(de_config)
model = cls(splayer, encoder, assigner, phone_size, decoder)
return model