Transformers
PyTorch
swinv2
Inference Endpoints
Edit model card

The Base model of the Poseidon family with approx. 158M parameters, see https://arxiv.org/abs/2405.19101.

It is trained on a computational grid of size 128 x 128 and expects density, velocity in horizontal direction, velocity in vertical direction, and pressure as inputs.

This is a research model.

Downloads last month
5
Unable to determine this model’s pipeline type. Check the docs .

Datasets used to train camlab-ethz/Poseidon-B

Collection including camlab-ethz/Poseidon-B