인코더와 어텐션 디코더를 이용하여 모델을 훈련시킵니다.
코드 10-39 어텐션 디코더 모델 훈련
import time
embed_size = 256
hidden_size = 512
num_layers = 1
input_size = input_lang.n_words
output_size = output_lang.n_words
encoder1 = Encoder(input_size, hidden_size, embed_size, num_layers)
attn_decoder1 = AttnDecoderRNN(hidden_size, output_size, dropout_p=0.1).to(device)
print(encoder1)
print(attn_decoder1)
attn_model = trainIters(encoder1, attn_decoder1, 75000, print_every=5000, plot_every=100, learning_rate=0.01) ------ 인코더와 어텐션 디코더를 이용한 모델 생성