esm2_t33_650M_UR50D-10k-uniprot-classification

This model is a fine-tuned version of facebook/esm2_t33_650M_UR50D on the None dataset. It achieves the following results on the evaluation set:

• Loss: 0.5318
• Accuracy: 0.8818

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 2e-05
• train_batch_size: 8
• eval_batch_size: 8
• seed: 42
• gradient_accumulation_steps: 16
• total_train_batch_size: 128
• optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• lr_scheduler_type: linear
• num_epochs: 7

Training results

Training Loss	Epoch	Step	Validation Loss	Accuracy
No log	0.9895	65	1.5025	0.6391
No log	1.9943	131	0.9955	0.7840
No log	2.9990	197	0.7501	0.8372
No log	3.9886	262	0.6199	0.8650
No log	4.9933	328	0.5642	0.8743
No log	5.9981	394	0.5376	0.8811
No log	6.9267	455	0.5318	0.8818

Framework versions

• Transformers 4.40.2
• Pytorch 2.3.0+cu121
• Datasets 2.19.1
• Tokenizers 0.19.1