2

I'm trying to optimize a BERT model with SageMaker Neo that is pre-trained using PyTorch, but I'm getting the following error:

ClientError: InputConfiguration: Exactly one .pt or .pth file is allowed for PyTorch models.

My model.tar.gz is only having the model.pth file in the root.

What am I missing here?

juvchan
  • 6,113
  • 2
  • 22
  • 35

0 Answers0