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

Question

我正在下载pre trained model，googlenetinception V1从视频帧中提取特征。然后是几个可训练变量。我有两个疑问需要解决。在

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

1. 在保存程序.还原（sess，checkpoint_文件）为预先训练的模型返回NoneType

Answer 1

TensorFlowSaver在构造时接受Variable列表来决定要还原或保存哪个Variable，默认情况下它将在默认图中找到所有变量。在

并且optimizer操作符也接受Variable列表来告诉训练时要优化哪些变量，并且它还默认地在默认图中找到所有的{}s

您可以定义2个Saver，一个用于加载InceptionV1Variables，另一个用于您自己的Variables

当构造optimizer时，你可以将你自己的Variable传递给参数var_list，这样列车只更新你自己的Variable，而不是{}Variable