renziver

**Describe the bug** I'm encountering an OOM error while using the Hugging Face Estimator API while following the [Spot Instance training sample](https://github.com/huggingface/notebooks/blob/main/sagemaker/05_spot_instances/sagemaker-notebook.ipynb) which basically needs a way to resume training...

### Willingness to contribute No. I cannot contribute a bug fix at this time. ### Feathr version 1.0.0 ### System information - Mac OS Ventura 13.5.1 - Python 3.8 ###...