|
Algorithm 1 Original Stepwise Weight Pruning Algorithm (SWPA) [43] |
-
Input: training data , training labels Y, base network , Drop-in Layer W, Step Counter . Selection factor
for in do
if > 1 then
k
Sort the weights W of the Drop-in Layer based on their absolute value.
Set the least k of them to 0.
Train the base network on O
Take the features corresponding to the top f fraction of the weights in W based on their absolute value and train them on the base network.
|