Request to abstract out part of training.lua so that extensions do not need to copy that part of the code.