Text-Classification icon indicating copy to clipboard operation
Text-Classification copied to clipboard

怎么存模型呢

Open Fino2020 opened this issue 3 years ago • 0 comments

请问这个模型应该怎么存,我想写一个单独的预测函数,在训练位置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'))

但是报错如下: QQ截图20220720094928 请问是什么问题呢

Fino2020 avatar Jul 20 '22 01:07 Fino2020