File size: 409 Bytes
b2cc771 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import apebench
import jax.numpy as jnp
for scenario in [
"phy_aniso_diff",
"diff_burgers", # Uses the two-channel version in 2D
"diff_ks",
"phy_kolm_flow",
"phy_gs_type",
]:
scene = apebench.scenarios.scenario_dict[scenario](
num_spatial_dims=2,
)
ref_trj = scene.get_ref_sample_data()
jnp.save(f"ref_sample_rollouts/{scene.get_scenario_name()}.npy", ref_trj)
|