Update train.py

Author: ujjax

Diff for: train.py

@@ -184,16 +184,16 @@ def main(_):
 	data_generator = prepare_data.get_batches(data, batch_size)
 
 	hidden_size = 256
-	encoder1 = EncoderRNN(input_lang.n_words, hidden_size)
-	attn_decoder1 = AttnDecoderRNN(hidden_size, output_lang.n_words, dropout_p=0.1)
+	encoder1 = Encoder(input_lang.n_words, hidden_size)
+	decoder1 = Decoder(hidden_size, output_lang.n_words, dropout_p=0.1)
 
 
 	if use_cuda:
 	    encoder1 = encoder1.cuda()
-	    attn_decoder1 = attn_decoder1.cuda()
+	    attn_decoder1 = decoder1.cuda()
 
-	trainIters(encoder1, attn_decoder1, 75000, print_every=5000)
+	trainIters(encoder1, decoder1, 75000, print_every=5000)
 
 
 if __name__ == '__main__':
-	main()
+	main()