Skip to content

G
gm_strategy_cvr
Commit ea0906e4 authored by 赵威's avatar 赵威
Browse files
Options

update save path

parent 4bf48ce4
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
src/train_diary.py
View file @ ea0906e4
...@@ -83,8 +83,9 @@ def main(): ...@@ -83,8 +83,9 @@ def main():
# tf.saved_model.load # tf.saved_model.load
# save_path = get_essm_model_save_path("diary") # TODO
# print("load path: " + save_path) save_path = get_essm_model_save_path("diary")
print("load path: " + save_path)
predict_fn = tf.contrib.predictor.from_saved_model(save_path) predict_fn = tf.contrib.predictor.from_saved_model(save_path)
device_dict = device_fe.get_device_dict_from_redis() device_dict = device_fe.get_device_dict_from_redis()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment