-
Notifications
You must be signed in to change notification settings - Fork 194
Open
Description
请问这个模型应该怎么存,我想写一个单独的预测函数,在训练位置saver.save(sess, '../save_model/atttn_lstm_hierarchical/')
在预测的地方加载模型的时候使用
with tf.Session()as sess:
saver = tf.train.import_meta_graph('./save_model/atttn_lstm_hierarchical/-1000.meta')
saver.restore(sess, tf.train.latest_checkpoint('./save_model/atttn_lstm_hierarchical'))
Metadata
Metadata
Assignees
Labels
No labels
