saving checkpoint 1
saving checkpoint 2