Add architecture to save Learner model