Pruning in Keras

前端 未结 4 855
梦毁少年i
梦毁少年i 2021-01-04 00:54

I\'m trying to design a neural network using Keras with priority on prediction performance, and I cannot get sufficiently high accuracy by further reducing the number of lay

4条回答
  •  清歌不尽
    2021-01-04 01:52

    Take a look at Keras Surgeon: https://github.com/BenWhetton/keras-surgeon

    I have not tried it myself, but the documentation claims that it has functions to remove or insert nodes.

    Also, after looking at some papers on pruning, it seems that many researchers create a new model with less channels (or less layers), and then copy the weights from the original model to the new model.

提交回复
热议问题