tensorflow使用joblib保存与加载模型
1.tensorflow使用joblib保存模型:
import joblib
joblib.dump(params, file_name)
如图:
2.tensorflow使用joblib加载模型:
以一般的思路,使用如下代码是可以的:
import joblib
Policy = joblib.load(args.load_path)
然而报错:_pickle.UnpicklingError: invalid load key, 'x'.
需要写做:
import joblib
import tensorflow as tf
tf.reset_default_graph()
with tf.Session() as sess:
data = joblib.load(args.load_path)
policy = data['policy']