
迁移学习
还是少年呀
就读于电子科技大学计算机学院
展开
-
tensorflow fine_tune已训练模型的部分参数
由于很多时候我们在一个新的网络中只会用到一个已训练模型的部分参数,即迁移学习。 那么,如何加载已训练模型的部分参数到当前网络。 一、当前网络加载已训练模型相同name scope的变量 方法1. 手动构建与预训练一样的部分图 将需要fine tune的变量的name scope命名为与模型中的name scope相同,然后使用如下代码将模型参数加载到当前网络。 tf.train.Saver([va...原创 2019-08-28 20:04:51 · 1489 阅读 · 0 评论 -
npy文件读取与fine tune
npy文件是numpy专用的二进制文件,npy内容由一个字典组成,字典中的每一个键对应一层网络模型参数(包括权重w和偏置b)。 对于某些网络模型参数保存为.npy形式的文件,我们需要读取这些文件的参数,并fine tune到当前网络相应的变量。 def load_initial_weights(session): # Load the weights into memory weigh...原创 2019-08-28 21:37:26 · 712 阅读 · 0 评论