What’s preprossing? I’m sorry but I can’t debug your entire repository. If you have a simple script that reproduces the error, I can look into it.
There is no optimal method for accuracy and/or speed. Also, you cannot expect faster inference speed using pytorch pruning at the moment. The feature is experimental and not powered by a fast sparse linear algebra library yet.