update model card README.md

README.md

@@ -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 an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.4986
+- Loss: nan
 
 ## Model description
 
@@ -38,21 +38,18 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 8
+- num_epochs: 5
 - mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss |
-|:-------------:|:-----:|:----:|:---------------:|
-| 3.2072        | 1.0   | 100  | 2.6362          |
-| 2.5623        | 2.0   | 200  | 2.5323          |
-| 2.5298        | 3.0   | 300  | 2.5733          |
-| 2.5275        | 4.0   | 400  | 2.5487          |
-| 2.4336        | 5.0   | 500  | 2.5314          |
-| 2.5169        | 6.0   | 600  | 2.5311          |
-| 2.4437        | 7.0   | 700  | 2.3698          |
-| 2.4303        | 8.0   | 800  | 2.3818          |
+| Training Loss | Epoch | Step  | Validation Loss |
+|:-------------:|:-----:|:-----:|:---------------:|
+| 2.5245        | 1.0   | 10000 | nan             |
+| 2.5037        | 2.0   | 20000 | nan             |
+| 2.4967        | 3.0   | 30000 | nan             |
+| 2.4983        | 4.0   | 40000 | nan             |
+| 2.4926        | 5.0   | 50000 | nan             |
 
 
 ### Framework versions