Checkpoints and disk storage

[shainaraza]

thanks , it solved my issue, esp working with ’ load_best_model_at_end’