更新文档

Author: liuxinwei

doc/tutorials/IRModule.ipynb

@@ -98,7 +98,7 @@
     "# Create a dummy model\n",
     "class TorchModel(nn.Module):\n",
     "    def __init__(self):\n",
-    "        super(TorchModel, self).__init__()\n",
+    "        super().__init__()\n",
     "        self.fc1 = nn.Linear(784, 256)\n",
     "        self.relu1 = nn.ReLU()\n",
     "        self.fc2 = nn.Linear(256, 10)\n",
@@ -175,7 +175,7 @@
     "\n",
     "class RelaxModel(nn.Module):\n",
     "    def __init__(self):\n",
-    "        super(RelaxModel, self).__init__()\n",
+    "        super().__init__()\n",
     "        self.fc1 = nn.Linear(784, 256)\n",
     "        self.relu1 = nn.ReLU()\n",
     "        self.fc2 = nn.Linear(256, 10)\n",

doc/tutorials/demo.ipynb

@@ -64,7 +64,7 @@
     "\n",
     "class MLPModel(nn.Module):\n",
     "    def __init__(self):\n",
-    "        super(MLPModel, self).__init__()\n",
+    "        super().__init__()\n",
     "        self.fc1 = nn.Linear(784, 256)\n",
     "        self.relu1 = nn.ReLU()\n",
     "        self.fc2 = nn.Linear(256, 10)\n",
@@ -121,7 +121,8 @@
     }
    ],
    "source": [
-    "mod, param_spec = MLPModel().export_tvm(\n",
+    "model = MLPModel()\n",
+    "mod, param_spec = model.export_tvm(\n",
     "    spec={\"forward\": {\"x\": nn.spec.Tensor((1, 784), \"float32\")}}\n",
     ")\n",
     "mod.show()"
@@ -147,7 +148,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -156,7 +157,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [
     {
@@ -273,15 +274,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "[[24572.564 24926.229 25577.898 24807.32  25382.205 25409.445 24776.041\n",
-      "  26966.22  25864.61  24976.56 ]]\n"
+      "[[25814.24  25661.46  23081.768 25914.645 25132.182 26040.564 24963.717\n",
+      "  25476.984 24674.125 24782.957]]\n"
      ]
     }
    ],
@@ -350,7 +351,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.4"
+   "version": "3.12.2"
   }
  },
  "nbformat": 4,