File size: 2,220 Bytes
869c6bd
 
 
 
 
 
e8783e3
 
869c6bd
 
 
e8783e3
 
 
 
 
 
 
 
 
 
 
cb6ddde
 
 
 
 
 
 
869c6bd
e8783e3
 
 
 
 
cb6ddde
 
 
e8783e3
 
 
 
 
 
 
869c6bd
cb6ddde
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
---
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.