Added functionality for autosaving the collected statistics about the training / testing