Add intermediate checkpoints (epochs 1-9) and update model card

README.md

@@ -26,7 +26,7 @@ This model is a **masked autoencoder (MAE)**: the encoder sees the unmasked patc
 | Mask spatial radius `r` | one channel |
 | Mask temporal length `L` | 16 patches |
 | Masker parameter `pct_unmasked` | 0.45 |
-| Checkpoint | epoch 10 of 10 (`v9`, the one evaluated in the paper) |
+| Checkpoint | end of epoch 10 of 10 (the one evaluated in the paper); epochs 1–9 in `epoch_01/` … `epoch_09/` |
 | Encoder parameters | 12.69 M |
 | Training run | [`46gmjq13`](https://wandb.ai/pierregtch/chan-inv-clf/runs/46gmjq13) |
 
@@ -38,6 +38,8 @@ The paper recommends r = 9 cm, L = 2: see [`eeg-fm-masking_mae_r9cm_L2`](https:/
   `model.*`), i.e. exactly the tensors loaded for the downstream evaluation of the paper.
   The MAE decoder is
   not included; for JEPA the published weights are the student encoder, as evaluated in the paper.
+* `epoch_01/` … `epoch_09/`: `model.safetensors` of the intermediate checkpoints (end of epochs
+  1 to 9 of the same run), with the same tensors and key names; they use the same `config.json`.
 * `config.json`: the keyword arguments of `ContextualEncoderBenchmarkWrapper` (architecture +
   input scaling). Pass it unchanged as `model_kwargs`.
 * `metadata.json`: masking parameters, training-run id, checkpoint epoch/step.
@@ -91,6 +93,17 @@ backbone = PretrainedBackbone(
 )
 ```
 
+### Intermediate checkpoints (epochs 1–9)
+
+The end-of-epoch checkpoints of the same run are in the subfolders `epoch_01/` … `epoch_09/`
+(the final epoch 10 is the `model.safetensors` at the root). Build `model` as above, then:
+
+```python
+epoch = 5  # 1 to 9
+weights = load_file(hf_hub_download(repo, "model.safetensors", subfolder=f"epoch_{epoch:02d}"))
+model.load_state_dict(weights, strict=False)
+```
+
 ## Downstream results (OpenEEGBench, frozen encoder + ridge probe)
 
 Frozen encoder, ridge regression/classification on the flattened contextual features,

epoch_01/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b5d0ebf384c1e4754934d3f82f0b6270488a91a6d1b6ac07ffecb19cc228924e
+size 50771544

epoch_02/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97f43e0e479d2756cea3c6956639a530c3f0b5c996d60b48ef144a24769de626
+size 50771544

epoch_03/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:456f7474e2e5955edab6e83b4a613e8de562c6b93e50d613c897450a855d9425
+size 50771544

epoch_04/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a8701b7d30880f2aef78f633c3f60a06c1937167a6c1115e4a684a46b72f6235
+size 50771544

epoch_05/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ef86363c6e973263df4679556b0659ca432ca03d2da5bb41074be1c3c956701b
+size 50771544

epoch_06/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:746b6893384b2118c310d7460dd97f014256ec9b7740c6cabcf1ccf1c6f4478f
+size 50771544

epoch_07/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:accb609d7dcc02c8eeb976c2b0e547b340662690b5def0b4e376f234fb077306
+size 50771544

epoch_08/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:625ed12c923adf716f15a74beb383b1dc9934fa042a200881cb6ab237f8d2421
+size 50771544

epoch_09/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:79a7c48884845437b226536d95e471d4de5a5256579928d20abdb9503e081dae
+size 50771544