Aayan Mishra commited on
Upload folder using huggingface_hub
Browse files- checkpoint/config.json +14 -0
- checkpoint/metadata.json +6 -0
- checkpoint/pytorch_model.bin +3 -0
- checkpoint/training_state.pt +3 -0
checkpoint/config.json
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"esm_checkpoint": "esm2_t36_3B_UR50D",
|
| 3 |
+
"freeze_backbone": true,
|
| 4 |
+
"unfreeze_last_n_layers": 0,
|
| 5 |
+
"num_distance_bins": 64,
|
| 6 |
+
"pairwise_dim": 128,
|
| 7 |
+
"head_hidden_dim": 128,
|
| 8 |
+
"pairwise_chunk_size": 64,
|
| 9 |
+
"pairwise_checkpointing": false,
|
| 10 |
+
"predict_confidence": true,
|
| 11 |
+
"max_len": 512,
|
| 12 |
+
"_class": "ConstraintPredictor",
|
| 13 |
+
"_note": "Backbone weights not included; cfg.esm_checkpoint identifies backbone."
|
| 14 |
+
}
|
checkpoint/metadata.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 19,
|
| 3 |
+
"global_step": 16740,
|
| 4 |
+
"best_val_loss": 2.8557200742804487,
|
| 5 |
+
"is_best": true
|
| 6 |
+
}
|
checkpoint/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5a29643dfb88a9014bba2307fab2c8e238b887995dab1043dce9a5d16a568dae
|
| 3 |
+
size 5353655
|
checkpoint/training_state.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e99bc8b5683788f638bc26221247ea68158067d0fc76836fb1ede6db9c47f909
|
| 3 |
+
size 8082525
|