将预先训练的模型变量设置为不可训练的初始空间1

Question

我正在下载预训练模型，谷歌网络盗梦空间V1从视频帧中提取功能。然后是几个可训练变量。我有两个疑问要解决。

1. tf.trainable_variables()包含预训练模型的变量列表

​

​

1. saver.restore(sess，checkpoint_file)为预训练模型返回NoneType

Answer 1

TensorFlow Saver在构造要恢复或保存哪个Variable时接受Variable列表，默认情况下，它将在默认图形中找到所有变量。

optimizer操作符也接受Variable列表来判断训练时要优化哪些变量，它还默认查找默认图形中的所有Variable；

您可以定义两个Saver，一个用于加载InceptionV1 Variables，一个用于您自己的Variables。

在构造optimizer时，您可以将自己的Variable传递给参数var_list，因此火车只更新您自己的Variable，而不是InceptionV1 Variable。