Make it possible to gracefully abort the training process

[lecode-official]

When hitting Ctrl-C, the application should not crash with an exception, but the training should be aborted gracefully. Ideally, the latest model, pruning mask, and training statistics should be written to file.