From 8c64319272f7660a57c491d823384130823900fe Mon Sep 17 00:00:00 2001 From: "S.Cao" Date: Sat, 20 Jul 2024 14:21:30 -0500 Subject: [PATCH] fixed imports in examples --- examples/ex2_FNO3d_train_normalized.ipynb | 10 +++++----- examples/ex2_SFNO_train.ipynb | 4 +++- examples/ex2_SFNO_train_fnodata.ipynb | 2 ++ 3 files changed, 10 insertions(+), 6 deletions(-) diff --git a/examples/ex2_FNO3d_train_normalized.ipynb b/examples/ex2_FNO3d_train_normalized.ipynb index e908f8a..9093d76 100644 --- a/examples/ex2_FNO3d_train_normalized.ipynb +++ b/examples/ex2_FNO3d_train_normalized.ipynb @@ -32,12 +32,12 @@ "\n", "import torch\n", "\n", - "from fno.pipeline import *\n", - "from fno.datasets import *\n", - "from fno.losses import SobolevLoss\n", + "from sfno.pipeline import *\n", + "from sfno.datasets import *\n", + "from sfno.losses import SobolevLoss\n", "\n", - "from fno.fno3d import FNO3d\n", - "from fno.visualizations import plot_contour_trajectory\n", + "from sfno.fno3d import FNO3d\n", + "from sfno.visualizations import plot_contour_trajectory\n", "from torch.utils.data import DataLoader\n", "\n", "from fno.utils import get_seed\n", diff --git a/examples/ex2_SFNO_train.ipynb b/examples/ex2_SFNO_train.ipynb index 6bb4e03..15bdf6d 100644 --- a/examples/ex2_SFNO_train.ipynb +++ b/examples/ex2_SFNO_train.ipynb @@ -12,8 +12,10 @@ "import torch\n", "from sfno.utils import get_num_params, get_seed\n", "from sfno.pipeline import *\n", - "from sfno.visualizations import plot_contour_trajectory\n", + "from sfno.datasets import *\n", + "from sfno.losses import *\n", "from sfno.sfno import SFNO\n", + "from sfno.visualizations import plot_contour_trajectory\n", "from torch.utils.data import DataLoader\n", "get_seed(1127825, printout=False)\n", "\n", diff --git a/examples/ex2_SFNO_train_fnodata.ipynb b/examples/ex2_SFNO_train_fnodata.ipynb index a6534c1..9f8cddc 100644 --- a/examples/ex2_SFNO_train_fnodata.ipynb +++ b/examples/ex2_SFNO_train_fnodata.ipynb @@ -22,6 +22,8 @@ "import torch\n", "from sfno.utils import get_num_params, get_seed\n", "from sfno.pipeline import *\n", + "from sfno.datasets import *\n", + "from sfno.losses import *\n", "from sfno.sfno import SFNO\n", "from sfno.visualizations import plot_contour_trajectory\n", "from torch.utils.data import DataLoader\n",