Sagemaker huggingface estimator tries to import tensorflow when pytorch is defined

marinone94 · August 19, 2022, 1:33pm

I have problems running training jobs on sagemaker using the Huggingface estimator because it tries to import tensorflow even though pytorch is the defined framework. Did anyone face the same issue? Issue on Github contains all the details.

https://github.com/aws/sagemaker-python-sdk/issues/3319

Topic		Replies	Views
Huggingface / Pytorch versions on Sagemaker Amazon SageMaker	9	3350	December 20, 2022
Package errors running huggingface estimator on sagemaker Beginners	1	736	February 9, 2023
Huggingface SageMaker instance_type on example code Beginners	0	264	July 14, 2022
How to use albumentations in train.py in AWS Sagemaker? Beginners	0	438	March 7, 2023
Gated Model access error during training on SageMaker Beginners	1	556	September 4, 2023

Sagemaker huggingface estimator tries to import tensorflow when pytorch is defined

Related Topics