def __init__(self, voc_size, hidden_dim):
super(Seq2Seq, self).__init__()
self.encoder = Encoder.EncoderRNN(input_size=voc_size,
hidden_size=hidden_dim)
self.decoder = Decoder.DecoderRNN(hidden_size=hidden_dim,
output_size=voc_size)
