search

mathmanu / caffe-jacinto

This repository has moved. The new link can be obtained from https://github.com/TexasInstruments/jacinto-ai-devkit
116 stars 35 forks source link

When finetune with sparsity, how to exclude specific layer ? #16

Open ianz27 opened 6 years ago

ianz27 commented 6 years ago

I want to exclude some specific layer (for example by layer's name). How to config params or change code ?

mathmanu commented 6 years ago

Currently this is not supported via a parameter. But you cn easily do it via code changes.

You can find the function FindAndApplyChannelThresholdNet() in net.pp. Inside this function you can find the the loop that iterates through the layers in the network. You can insert a check for the layer name that you with to exclude and continue.