Add Training model objects to the object graph