Efficiently fine-tune the ESM-2 protein language model with Amazon SageMaker
AWS Machine Learning
MARCH 6, 2024
In our training script, we subclass the Trainer class from transformers with a WeightedTrainer class that takes class weights into account when calculating cross-entropy loss. Clean up If you’re following along in your own AWS account, delete the any real-time inference endpoints and data you created to avoid further charges.
114 
Let's personalize your content