| --- |
| library_name: pytorch |
| --- |
| |
| # DiffPhyCon Jellyfish Checkpoints |
|
|
| Checkpoint archive for the 2D Jellyfish DiffPhyCon reproduction and Flow |
| Matching experiments. |
|
|
| ## Contents |
|
|
| - `official/joint_full/model-100.pt`: released joint DDPM checkpoint. |
| - `official/w_full/model-50.pt`: released opening-angle prior checkpoint. |
| - `official/force_surrogate_model/force_model_epoch_9.pth`: released force surrogate. |
| - `official/boundary_updater/boundary_updater_epoch_9.pth`: released boundary updater. |
| - `flow_matching/joint_7ch/joint_step10000.pt` through |
| `joint_step160000.pt`: the complete saved 7-channel Jellyfish Flow Matching |
| joint-model training series, at 10,000-step intervals. |
| - `flow_matching/joint_7ch/joint_latest.pt`: the latest saved joint checkpoint |
| (corresponding to the 160,000-step training state). |
| - `flow_matching/joint_7ch/SHA256SUMS`: integrity checksums for the Flow |
| Matching joint checkpoints. |
| - `flow_matching/prior_7ch/prior_step10000.pt` through |
| `prior_step70000.pt`: the complete saved 7-channel Jellyfish Flow Matching |
| control-prior training series, at 10,000-step intervals. |
| - `flow_matching/prior_7ch/prior_latest.pt`: the latest saved prior checkpoint |
| (corresponding to the 70,000-step training state). |
| - `flow_matching/prior_7ch/SHA256SUMS`: integrity checksums for the Flow |
| Matching prior checkpoints. |
|
|
| The Flow Matching files retain the model, EMA model, optimizer, scheduler, |
| configuration, and training-step state needed for inference and training |
| resumption. Intermediate checkpoints are intentionally retained for exact |
| experiment recovery and checkpoint-selection studies. |
|
|
| Do not confuse the official DDPM control prior at |
| `official/w_full/model-50.pt` with the separately trained Flow Matching prior |
| under `flow_matching/prior_7ch/`. |
|
|
| ## Provenance |
|
|
| The files under `official/` were released by the authors of |
| [DiffPhyCon](https://github.com/AI4Science-WestlakeU/diffphycon). The files |
| under `flow_matching/` are checkpoints produced by the Flow Matching |
| implementation in this project. |
|
|
| This repository is published for reproducible research with explicit separation |
| between the original DiffPhyCon releases and the Flow Matching checkpoints |
| produced by this project. |
|
|
