Hello! I am new to Huggingface and Sagemaker, and I am trying to fine-tune a Huggingface model in Sagemaker using the HuggingFace Estimator.
train.py file, I need to import and use the
albumentations library for data augmentation. I have already installed
albumentations in Sagemaker notebook using pip install. However, when I run
train.py, it cannot recognize that the
albumentations library is installed.
I have looked for solutions to this issue in the forums, but haven’t been able to find a solution that works for me. Can anyone please help me with this issue? Thanks in advance!
Similar question in the forum fyi: