Added in saving the final model to the training script.
Added in saving the final model to the training script.

The script now handles CTRL-C's by the user, and saves the final model
when training is finished. Also, some more print messages were added to
let the user know how many files were found in the input directory.

bperez77/ensemble_colorization,bperez77/ensemble_colorization,bperez77/ensemble_colorization