Updated training function to work for lazy loading or preloading of t…#205

gperdrizet merged 1 commit intomaingperdrizet/FSA_devops:mainfrom

devgperdrizet/FSA_devops:devCopy head branch name to clipboard