BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #29808同步于 2018/6/1
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

tensorflow怎么保存剪枝后的模型

w86568376
2018/6/1镜像同步7 回复
有在做神经网络剪枝的大佬吗。。 我用tf.assign函数把卷积核参数张量替换成剪枝后的参数张量,但是assign函数只改变张量的具体值,不改变张量的shape属性。这会导致保存ckpt文件,下次加载回来的时候参数形状对不上。 举个例子,本来卷积核参数形状是[3,3,3,64],剪枝后是[3,3,3,45],虽然张量里的值确实变了(打印出来看过),但是保存ckpt文件时shape属性仍然是[3,3,3,64],导致加载参数的时候形状对不上,加载失败。 有在做剪枝方面的大佬吗,求指点一下应该怎么做?或者是不是我得换一个框架了,有什么框架支持剪枝比较好的吗?谢谢!
订阅后,新回复会通过你的通知中心匿名送达。
7 条回复
KillerDeath机器人#1 · 2018/6/1
活捉
KillerDeath机器人#2 · 2018/6/1
如果剪枝之后shape不变,那模型大小岂不是也没减少?
w86568376机器人#3 · 2018/6/1
参数实际是变了…但是张量的shape属性没变,个人认为是个bug 【 在 KillerDeath (KillerDeath) 的大作中提到: 】 : 如果剪枝之后shape不变,那模型大小岂不是也没减少?
w86568376机器人#4 · 2018/6/1
相当于你的个人档案里身高1.9m,但是实际上你只有1.75m! 【 在 w86568376 (然后就没有然后) 的大作中提到: 】 : 参数实际是变了…但是张量的shape属性没变,个人认为是个bug
Viredery机器人#5 · 2018/6/1
他只是把剪枝后的那些层的参数置为0了,你得手动做内存重排。。
jaegerstar机器人#6 · 2018/6/5
今晚来我宿舍我们讨论一下
w86568376机器人#7 · 2018/6/5
已解决 【 在 jaegerstar (jaegerstar) 的大作中提到: 】 : 今晚来我宿舍我们讨论一下