Caleb Ellington
commited on
Commit
·
9625014
1
Parent(s):
beaed13
Update paths
Browse files- README.md +2 -2
- config.yaml +5 -5
README.md
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
datasets:
|
3 |
- Bo1015/ssp_q3
|
4 |
base_model:
|
5 |
-
- genbio-ai/
|
6 |
---
|
7 |
LoRA fine-tuned checkpoint for ssp_q3.
|
8 |
|
@@ -12,7 +12,7 @@ LoRA fine-tuned checkpoint for ssp_q3.
|
|
12 |
from huggingface_hub import snapshot_download
|
13 |
from pathlib import Path
|
14 |
|
15 |
-
model_name = "genbio-ai/
|
16 |
genbio_models_path = Path.home().joinpath('genbio_models', model_name)
|
17 |
genbio_models_path.mkdir(parents=True, exist_ok=True)
|
18 |
snapshot_download(repo_id=model_name, local_dir=genbio_models_path)
|
|
|
2 |
datasets:
|
3 |
- Bo1015/ssp_q3
|
4 |
base_model:
|
5 |
+
- genbio-ai/AIDO.Protein-16B
|
6 |
---
|
7 |
LoRA fine-tuned checkpoint for ssp_q3.
|
8 |
|
|
|
12 |
from huggingface_hub import snapshot_download
|
13 |
from pathlib import Path
|
14 |
|
15 |
+
model_name = "genbio-ai/AIDO.Protein-16B-ssp_q3"
|
16 |
genbio_models_path = Path.home().joinpath('genbio_models', model_name)
|
17 |
genbio_models_path.mkdir(parents=True, exist_ok=True)
|
18 |
snapshot_download(repo_id=model_name, local_dir=genbio_models_path)
|
config.yaml
CHANGED
@@ -36,14 +36,14 @@ trainer:
|
|
36 |
logger:
|
37 |
class_path: lightning.pytorch.loggers.WandbLogger
|
38 |
init_args:
|
39 |
-
name:
|
40 |
save_dir: lightning_logs/protein
|
41 |
version: null
|
42 |
offline: false
|
43 |
dir: null
|
44 |
id: null
|
45 |
anonymous: null
|
46 |
-
project:
|
47 |
log_model: false
|
48 |
experiment: null
|
49 |
prefix: ''
|
@@ -77,7 +77,7 @@ trainer:
|
|
77 |
log_weight_decay: false
|
78 |
- class_path: lightning.pytorch.callbacks.ModelCheckpoint
|
79 |
init_args:
|
80 |
-
dirpath:
|
81 |
filename: best_val:{epoch}-{val_accuracy:.3f}
|
82 |
monitor: val_accuracy
|
83 |
verbose: false
|
@@ -149,7 +149,7 @@ model:
|
|
149 |
dropout: 0.1
|
150 |
dropout_in_middle: true
|
151 |
backbone:
|
152 |
-
class_path: modelgenerator.backbones.
|
153 |
init_args:
|
154 |
from_scratch: false
|
155 |
max_length: 512
|
@@ -199,7 +199,7 @@ model:
|
|
199 |
data:
|
200 |
class_path: modelgenerator.data.SspQ3
|
201 |
init_args:
|
202 |
-
path:
|
203 |
batch_size: 1
|
204 |
max_length: 512
|
205 |
config_name: null
|
|
|
36 |
logger:
|
37 |
class_path: lightning.pytorch.loggers.WandbLogger
|
38 |
init_args:
|
39 |
+
name: ssp_AIDO.Protein-16B
|
40 |
save_dir: lightning_logs/protein
|
41 |
version: null
|
42 |
offline: false
|
43 |
dir: null
|
44 |
id: null
|
45 |
anonymous: null
|
46 |
+
project: xtrimo_benchmark_mgen_reproduction
|
47 |
log_model: false
|
48 |
experiment: null
|
49 |
prefix: ''
|
|
|
77 |
log_weight_decay: false
|
78 |
- class_path: lightning.pytorch.callbacks.ModelCheckpoint
|
79 |
init_args:
|
80 |
+
dirpath: logs/xtrimo_benchmark_mgen_reproduction/ssp_AIDO.Protein-16B
|
81 |
filename: best_val:{epoch}-{val_accuracy:.3f}
|
82 |
monitor: val_accuracy
|
83 |
verbose: false
|
|
|
149 |
dropout: 0.1
|
150 |
dropout_in_middle: true
|
151 |
backbone:
|
152 |
+
class_path: modelgenerator.backbones.aido_protein_16b
|
153 |
init_args:
|
154 |
from_scratch: false
|
155 |
max_length: 512
|
|
|
199 |
data:
|
200 |
class_path: modelgenerator.data.SspQ3
|
201 |
init_args:
|
202 |
+
path: proteinglm/ssp_q3
|
203 |
batch_size: 1
|
204 |
max_length: 512
|
205 |
config_name: null
|