How to resume the train?

[dengxiongshi]

@HankYe Hello! If the compress script is kill, I want to continu train the experiment. So, what can I do?

[HankYe]

Hello, thanks for ur attention to our work. To resume training the pruned model, you can add some codes to load the optimizer states of the pruned model after line 192. Meanwhile, the value of start epoch should be updated.