File size: 778 Bytes
293a990 d5ab9b5 293a990 d5ab9b5 | 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 | ---
license: mit
tags:
- pytorch-lightning
- neural-pde-solver
- operator-splitting
---
# DISCO model checkpoints
Pretrained checkpoints for the ICML 2026 paper *Test-time Generalization for
Physics through Neural Operator Splitting* (Serrano, Han, Oyallon, Ho, Morel).
| Equation | Checkpoint |
|---|---|
| Advection-diffusion (1D) | `advection-diffusion/last.ckpt` |
| Combined equation (1D) | `combined-equation/last.ckpt` |
| Reaction-diffusion (Gray-Scott, 2D) | `reaction-diffusion/last.ckpt` |
| Navier-Stokes / Euler (2D) | `navier-stokes/best-checkpoint.ckpt` |
Load with PyTorch Lightning:
```python
from train.train_generic import DISCOLitModule
model = DISCOLitModule.load_from_checkpoint("last.ckpt")
```
Code: https://github.com/LouisSerrano/disco-ttg
|