将预先训练的模型变量设置为不可训练的初始空间1
将预先训练的模型变量设置为不可训练的初始空间1
提问于 2017-08-31 04:36:36
我正在下载预训练模型，谷歌网络盗梦空间V1从视频帧中提取功能。然后是几个可训练变量。我有两个疑问要解决。
- tf.trainable_variables()包含预训练模型的变量列表
- saver.restore(sess，checkpoint_file)为预训练模型返回NoneType
回答 1
Stack Overflow用户
发布于 2017-08-31 04:51:02
TensorFlow Saver在构造要恢复或保存哪个Variable时接受Variable列表，默认情况下，它将在默认图形中找到所有变量。
optimizer操作符也接受Variable列表来判断训练时要优化哪些变量，它还默认查找默认图形中的所有Variable；
您可以定义两个Saver，一个用于加载InceptionV1 Variables，一个用于您自己的Variables。
在构造optimizer时，您可以将自己的Variable传递给参数var_list，因此火车只更新您自己的Variable，而不是InceptionV1 Variable。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/45973319
复制相关文章
相似问题
腾讯云开发者
