ReViT pretrained models

Pretrained checkpoints for ReViT: Rotational-equivariant Vision Transformers for Neural PDE Solvers. See the code at https://github.com/Howw-Way/ReViT.

Model	Channels	Grid	Patch	Architecture
P3D	4	96³	3	ReViT3D
MHD_64	7	64³	2	ReViT3D

Usage

from revit import load_pretrained
model = load_pretrained("P3D")        # or "MHD_64"

Each subfolder holds config.json (the model config), best_model.pth (state dict), and the original config.yaml for provenance.