move fft functions in equations to spectral

Author: scaomath

examples/ex2_FNO3d_train_normalized.ipynb

@@ -97,11 +97,11 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
-    "train_dataset = BochnerDatasetFixed(\n",
+    "train_dataset = SpatioTemporalDatasetFixedTime(\n",
     "    datapath=TRAIN_PATH,\n",
     "    fields=[\"vorticity\"],\n",
     "    n_samples=Ntrain,\n",
@@ -112,7 +112,7 @@
     "    out_steps=T,\n",
     "    T_start=T_start,\n",
     ")\n",
-    "test_dataset = BochnerDatasetFixed(\n",
+    "test_dataset = SpatioTemporalDatasetFixedTime(\n",
     "    datapath=VALID_PATH,\n",
     "    fields=[\"vorticity\"],\n",
     "    n_samples=Ntest,\n",
@@ -493,7 +493,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.9"
+   "version": "3.10.17"
   }
  },
  "nbformat": 4,

examples/ex2_SFNO_finetune_fnodata.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -14,11 +14,13 @@
     "import torch.nn.functional as F\n",
     "from torch.linalg import norm\n",
     "\n",
-    "from sfno.utils import get_seed\n",
-    "from sfno.pipeline import *\n",
-    "from sfno.visualizations import *\n",
-    "from sfno.sfno import SFNO\n",
-    "from sfno.finetune import OutConvFT\n",
+    "from fno.utils import get_seed\n",
+    "from fno.pipeline import *\n",
+    "from fno.visualizations import *\n",
+    "from fno.datasets import *\n",
+    "from fno.losses import *\n",
+    "from fno.sfno import SFNO\n",
+    "from fno.finetune import OutConvFT\n",
     "from torch.utils.data import DataLoader\n",
     "\n",
     "get_seed(42, printout=False)\n",
@@ -78,11 +80,11 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
-    "test_dataset = BochnerDataset(datapath=TEST_PATH, \n",
+    "test_dataset = SpatioTemporalDataset(datapath=TEST_PATH, \n",
     "                              fields=fields,\n",
     "                              n_samples=Ntest,\n",
     "                              steps=T,\n",