Hi @Michela
you can try running my script https://github.com/raghavgurbaxani/Quantization_Experiments/blob/master/try_pruning.py 13
based on this repository -

  
      
      GitHub
  
  
    

ZhangMingliangAI/EAST-Pytorch 9

This is a project of EAST text detector implemented by Pytorch - ZhangMingliangAI/EAST-Pytorch


  
  
    
    
  
  


pruning only the weight works, but as soon as I prune the bias as well it throws me the error