Added code to select the weights used for reinitialization, rather than using the weights from the initialization.

Selecting the weight is done using the parameter -o / --original-epoch. When providing this parameter with the value 0, you will still use the original initialization weights.

The first pruning iteration will train for trainepochs, while the other iterations will train for trainepochs - original-epoch