sosick377

thank you for sharing the your code. I have a question about "Get_Weight_Gradient" function in content aware pruning. You only consider the gradient of to_rgbs[-1] in rgb layers. (I mean...