Use the lazy=False option for the run method in the intermediate steps of the learning