Update README.md

README.md

@@ -6,16 +6,28 @@ 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 GAL4_YEAST. 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 GAL4_YEAST. 
 
 ### 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
-
+Anomino acid
 ### 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