pc2976
/

prot_bert-finetuned-sp6

generated_from_trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

pc2976 commited on Aug 4, 2022

Commit

0aea4e5

•

1 Parent(s): 43355f8

update model card README.md

Files changed (1) hide show

README.md +5 -5

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [Rostlab/prot_bert](https://huggingface.co/Rostlab/prot_bert) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4511
 ## Model description
@@ -32,7 +32,7 @@ More information needed
 ### Training hyperparameters
 The following hyperparameters were used during training:
-- learning_rate: 2e-05
 - train_batch_size: 32
 - eval_batch_size: 32
 - seed: 42
@@ -44,9 +44,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 63   | 0.4757          |
-| No log        | 2.0   | 126  | 0.4482          |
-| 0.502         | 3.0   | 189  | 0.4419          |
 ### Framework versions

 This model is a fine-tuned version of [Rostlab/prot_bert](https://huggingface.co/Rostlab/prot_bert) on the None dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.4070
 ## Model description
 ### Training hyperparameters
 The following hyperparameters were used during training:
+- learning_rate: 0.0002
 - train_batch_size: 32
 - eval_batch_size: 32
 - seed: 42
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
+| 0.5027        | 1.0   | 164  | 0.4666          |
+| 0.3927        | 2.0   | 328  | 0.4328          |
+| 0.3348        | 3.0   | 492  | 0.4072          |
 ### Framework versions