import pickle file = open('./Data/T2_data_train.pt', 'rb') df = pickle.load(file)[:120] German_corpus = Corpus() English_corpus = Corpus() German_corpus.load_vocab('./Parameters/German_corpus1k.pt') English_corpus.load_vocab('./Parameters/English_corpus1k.pt') German_vectorizer = WordsToVec(German_corpus, 150) English_vectorizer = WordsToVec(English_corpus, 150) German_vectorizer.load_model_state('./Parameters/German_vectors_150d') English_vectorizer.load_model_state('./Parameters/English_vectors_150d') my_translator = Translator(German_corpus, English_corpus, German_vectorizer, English_vectorizer, dec_GRU_count=2, enc_GRU_count=2, hidden_size=128, use_feedforward=True) # my_translator.train(df, epochs=50, batch_size=32, lr=0.001, lr_reduce=0.1) my_translator.load_model('./Parameters/translator_feedforwardatt_epoch50.pt') my_translator.reinforce_train(df, epochs=50, batch_size=32)