Updated Training Strategy classes