metadata
library_name: revit-pde
tags:
- pde
- equivariant
- vision-transformer
- neural-operator
ReViT pretrained models
Pretrained checkpoints for ReViT: Rotational-equivariant Vision Transformers for Neural PDE Solvers. See the code at https://github.com/tum-pbs/revit.
| Model | Channels | Grid | Patch | Architecture |
|---|---|---|---|---|
TCF |
4 | 96³ | 3 | ReViT3D |
MHD_64 |
7 | 64³ | 2 | ReViT3D |
Usage
from revit import load_pretrained
model = load_pretrained("TCF") # or "MHD_64"
Each subfolder holds config.json (the model config), best_model.pth
(state dict), and the original config.yaml for provenance.