tensorflow使用joblib保存与加载模型

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']