Update README.md

README.md

@@ -6,17 +6,35 @@ library_name: peft
 # Model Card for Model ID
 
 <!-- Provide a quick summary of what the model is/does. -->
-This model is used to predict mutation fitness score of protein DLG4_RAT. The dataset is from [SESNet: sequence-structure feature-integrated deep learning method for data-efficient protein engineering](https://jcheminf.biomedcentral.com/articles/10.1186/s13321-023-00688-x).
+This model is used to predict mutation fitness score of protein DLG4_RAT. 
 
 ### Task type
-
+protein level regression
 ### Dataset description
-
+The dataset is from [SESNet: sequence-structure feature-integrated deep learning method for data-efficient protein engineering](https://jcheminf.biomedcentral.com/articles/10.1186/s13321-023-00688-x).
 ### Model input type
-
+Amino acid sequence
 ### Performance
 
+
 ### LoRA config
+lora_dropout: 0.0
+lora_alpha: 16
+target_modules: ["query", "key", "value", "intermediate.dense", "output.dense"]
+modules_to_save: ["classifier"]
 
 ### Training config
+class: AdamW
+
+betas: (0.9, 0.98)
+
+weight_decay: 0.01
+
+learning rate: 1e-4
+
+epoch: 50
+
+batch size: 2
+
+precision: 16-mixed