model card

README.md

@@ -51,11 +51,17 @@ extras, not pieces you have to assemble.
 | Path in this repository | Size | Where it goes in the code repo |
 |---|---:|---|
 | **`poly4mer_pretrained_v3/poly4mer_v3.ckpt`** | 3.7 GB | `checkpoint/` — the one you need |
+| `poly4mer_pretrained_v3/property_heads_10folds/<p>/fold_{0..9}.ckpt` | 126 MB | `checkpoint/fire_property_heads/<p>/` |
+| `poly4mer_pretrained_v3/monotone_heads_10folds/<p>/fold_{0..9}_best_fp32.ckpt` | 63 MB | `poly4mer_monotone/checkpoint/<p>/last_two_linear/` |
 | `poly4mer_pretrained_v2/smi_ted_light/smi-ted-Light_40.pt` | 1.1 GB | `smi_ted_light/` |
-| `poly4mer_pretrained_v3/tig_10folds/fold_{0..9}_best_fp32.ckpt` | 32 MB | `poly4mer_monotone/checkpoint/tig/last_two_linear/` |
-| `poly4mer_pretrained_v3/phrr_10folds/fold_{0..9}_best_fp32.ckpt` | 32 MB | `poly4mer_monotone/checkpoint/phrr/last_two_linear/` |
-| `poly4mer_pretrained_v3/sea_10folds/fold_{0..9}.ckpt` | 32 MB | `checkpoint/fire_property_heads/sea/` |
-| `poly4mer_pretrained_v3/co_10folds/fold_{0..9}.ckpt` | 32 MB | `checkpoint/fire_property_heads/co/` |
+
+`<p>` is `tig`, `pkhrr`, `sea` or `co` for the per-fold heads, and `tig` or `phrr` for the
+monotone ones. The v2 release is under `poly4mer_pretrained_v2/`.
+
+The two head folders are optional. `property_heads_10folds/` holds the heads behind the
+cross-validation tables, needed only to compare your own re-training against ours.
+`monotone_heads_10folds/` holds the hard-monotone Tig and pHRR heads described below.
+Their folder names say which kind they are, so they cannot be mixed up.
 
 Only the checkpoint and the SMI-TED weights are needed for prediction, reconstruction or
 design.