Просмотр исходного кода

Merge pull request #566 from 372046933/patch-1

Update translate_sample.py
BO-YANG HSUEH 5 лет назад
Родитель
Сommit
2af70620bb
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      FasterTransformer/v2/sample/tensorflow/translate_sample.py

+ 2 - 2
FasterTransformer/v2/sample/tensorflow/translate_sample.py

@@ -24,7 +24,7 @@ from opennmt.decoders.self_attention_decoder import SelfAttentionDecoder
 from opennmt.inputters import WordEmbedder
 from opennmt.inputters import ExampleInputter
 
-def resotre_model_by_pkl(sess, variables):
+def restore_model_by_pkl(sess, variables):
     import pickle as pkl
     with open("model.pkl", 'rb') as model_file:
         model_dict = pkl.load(model_file)
@@ -226,7 +226,7 @@ if __name__ == "__main__":
         saver.restore(sess, "translation/ckpt/model.ckpt-500000")
         sess.run(tf.tables_initializer())
         sess.run(iterator.initializer)
-        resotre_model_by_pkl(sess, decoder_variables)
+        restore_model_by_pkl(sess, decoder_variables)
         
         iteration = 0
         while iteration < 3: