include proved model in one shot training