我使用生成已知四个文件(.data、.meta、ckpt、.index)的 Tensorflow 1.x 保存了经过训练的模型。我想在 Tensorflow 2.x 中加载模型并将其用于对我自己的数据集进行微调。下面是我用来访问 1.x 模型的代码。
import tensorflow as tf
tf.compat.v1.disable_eager_execution()
session = tf.compat.v1.Session()
saver = tf.compat.v1.train.import_meta_graph('NameOfModel.meta')
saver.restore(session,tf.train.latest_checkpoint('./'))
graph = tf.compat.v1.get_default_graph()
访问模型中的变量:
list_of_variables = tf.compat.v1.trainable_variables()
转换到 Tensorflow 2.x/Keras 以查看模型架构/摘要、继续训练和预测新数据集上的值的最简单方法是什么?