tensorflow 函数 slim

tensorflow在contrib下有一个函数slim,导入方式:

import tensorflow.contrib.slim as slim

我在用saver()读取之前模型的时候多次遇到了attempting to use uninitialized variable的问题,因此需要查看读取的变量。  

在存模型或读模型之前用slim可以去查看存了哪些变量

model_variables = slim.get_variables()
restore_variables = [var for var in model_variables]
for var in restore_variables:
    print(var.name)

slim的功能不止这个,实际上slim是一个简化的构建模型方式,tutorial:官方说明

猜你喜欢

转载自blog.csdn.net/thormas1996/article/details/81000655