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

Question

我正在下载pre trained model，Google Net Inception V1以从视频帧中提取功能。然后是几个可训练的变量。我有两个疑问需要解决。

1. tf.trainable_variables（）包括预训练模型的变量列表



1. saver.restore（sess，checkpoint_file）为预先训练的模型返回NoneType

Answer 1

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

optimizer运算符也接受Variable列表来告知在训练时要优化哪些变量，并且默认情况下还会在默认图表中找到所有Variable;

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

构建optimizer时，您可以将自己的Variable传递给参数var_list，这样火车就会更新您自己的Variable而不是InceptionV1 }} Variable s。