Модель переобучается при выводе модели NMT - PullRequest
0 голосов
/ 18 сентября 2018

Я изменил официальный ноутбук Google nmt на 3 файла: data_prep.py, train_model.py и inference.py.Когда я запускаю скрипт inference.py, модель переучивается с нуля, а не использует сохраненную модель.В чем может быть ошибка в коде и как я могу ее устранить?

Ниже приведен код inference.py

import tensorflow as tf
from train_model import inference, optimizer, encoder, decoder, inp_que, targ_ans, max_length_inp, max_length_targ, checkpoint

checkpoint_dir = './training_checkpoints'
# restoring the latest checkpoint in checkpoint_dir
checkpoint.restore(tf.train.latest_checkpoint(checkpoint_dir))

print(inference('hello', encoder, decoder, inp_que, targ_ans, max_length_inp, max_length_targ))

Если кто-то хочет взглянуть на репозиторий, он может получить к нему доступ.это здесь

...