Tensorflow: 如何加载神经网络部分参数

转载请标明出处:https://blog.csdn.net/SingingFisherhtml   需求以下:先用Resnet50在ImageNet上预训练,最后一层输出为类别数量,设为1000。而后将保存下来的参数迁移到PascalVoc上作调优训练。函数 问题:因为PascalVoc只有20类,因此Resnet50最后一层输出要改成20。此时直接用tf.train.Saver()的restor
相关文章
相关标签/搜索