Pruning a neural Network for faster training times