@NDugar I would rather recommend running your training using the HuggingFace estimator than the Notebook Instance. SageMaker will handle the management of the resources behind the scene and makes sure the training runs.