Jiawei Zhang commited on
Commit
5ae9b45
1 Parent(s): b71beca

Upload 5 files

Browse files
Files changed (4) hide show
  1. README.md +29 -3
  2. adapter_config.json +34 -0
  3. adapter_model.safetensors +3 -0
  4. metadata.json +3 -0
README.md CHANGED
@@ -1,3 +1,29 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ ---
4
+
5
+ **Base model:** [westlake-repl/SaProt_650M_AF2](https://huggingface.co/westlake-repl/SaProt_650M_AF2)
6
+
7
+ **Dataset:** This dataset contains 100K randomly generated mutants derived from the wild type EYFP protein. The number of samples for
8
+ training, validation and test is 100317, 5969 and 5968. 10% of double-site mutants and 10% of triple-site mutants were used for validation and test
9
+ respectively, and the remains for training. This model was trained by Jia Zheng's lab at Westlake University. The dataset will be released later by this team.
10
+
11
+ **Performance (on test set):** 0.94 Spearman's ρ
12
+
13
+ **LoRA config:**
14
+ - **r:** 8
15
+ - **lora_dropout:** 0.0
16
+ - **lora_alpha:** 16
17
+ - **target_modules:** ["query", "key", "value", "intermediate.dense", "output.dense"]
18
+ - **modules_to_save:** ["classifier"]
19
+
20
+ **Training config:**
21
+
22
+ - **optimizer:**
23
+ - **class:** AdamW
24
+ - **betas:** (0.9, 0.98)
25
+ - **weight_decay:** 0.01
26
+ - **learning rate:** 1e-4
27
+ - **epoch:** 20
28
+ - **batch size:** 64
29
+ - **precision:** 16-mixed
adapter_config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "/sujin/Models/SaProt/SaProt_650M_AF2",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layer_replication": null,
10
+ "layers_pattern": null,
11
+ "layers_to_transform": null,
12
+ "loftq_config": {},
13
+ "lora_alpha": 16,
14
+ "lora_dropout": 0.0,
15
+ "megatron_config": null,
16
+ "megatron_core": "megatron.core",
17
+ "modules_to_save": [
18
+ "classifier"
19
+ ],
20
+ "peft_type": "LORA",
21
+ "r": 8,
22
+ "rank_pattern": {},
23
+ "revision": null,
24
+ "target_modules": [
25
+ "value",
26
+ "key",
27
+ "output.dense",
28
+ "intermediate.dense",
29
+ "query"
30
+ ],
31
+ "task_type": "SEQ_CLS",
32
+ "use_dora": false,
33
+ "use_rslora": false
34
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74c5dd376ee3552f3fd9d57094cc15ae10182569f5a1d8471e03a6e4b5aa37ae
3
+ size 15502914
metadata.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "training_data_type": "AA"
3
+ }