Pruning while training

[codestar12]

Hello,

First of all thank you for building this library it is very helpful.

Looking through the examples I see that you tend to first train -> prune -> save -> load -> finetune

Is there any way to prune and continue training?

I think that would require making modifications to the optimizer right? but it should be possible.