Upload folder using huggingface_hub
Browse files- README.md +5 -0
- aptatrans_config.json +17 -0
- aptatrans_ft_pytorch_model.bin +3 -0
- lightning_last.ckpt +3 -0
README.md
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# AptaTrans finetuned checkpoint
|
| 2 |
+
|
| 3 |
+
- `pytorch_model.bin`: AptaTrans nn.Module state_dict (keys match pyaptamer AptaTrans)
|
| 4 |
+
- `aptatrans_config.json`: architecture+vocab config to reconstruct the model
|
| 5 |
+
- `lightning_last.ckpt`: original Lightning checkpoint (for debugging/resume)
|
aptatrans_config.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"n_apta_vocabs": 127,
|
| 3 |
+
"n_apta_target_vocabs": 344,
|
| 4 |
+
"apta_max_len": 275,
|
| 5 |
+
"n_prot_vocabs": 715,
|
| 6 |
+
"n_prot_target_vocabs": 585,
|
| 7 |
+
"prot_max_len": 867,
|
| 8 |
+
"in_dim": 128,
|
| 9 |
+
"n_encoder_layers": 6,
|
| 10 |
+
"n_heads": 8,
|
| 11 |
+
"conv_layers": [
|
| 12 |
+
3,
|
| 13 |
+
3,
|
| 14 |
+
3
|
| 15 |
+
],
|
| 16 |
+
"dropout": 0.1
|
| 17 |
+
}
|
aptatrans_ft_pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f98415ee1e8a31a9c861e975d248ec200c9527324f3c8713c22af776e037bfb4
|
| 3 |
+
size 25684159
|
lightning_last.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a6aa36c4a1e103154b26e7cbc92c53eb173ad18c5c4d52f1098ec987369c8890
|
| 3 |
+
size 74022321
|