Update translate_sample.py

372046933 · web-flow · commit fe2cef59f5b2 · 2020-06-19T17:47:47.000+08:00
fix typo
diff --git a/FasterTransformer/v2/sample/tensorflow/translate_sample.py b/FasterTransformer/v2/sample/tensorflow/translate_sample.py
@@ -24,7 +24,7 @@
 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 @@ def resotre_model_by_pkl(sess, variables):
         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:
