Merge branch 'main' of github.com:inoue0426/drGAT

inoue0426 · Aug 30, 2024 · 39ab42b · 39ab42b
2 parents edf589a + 5ca7597
commit 39ab42b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Tutorial.ipynb b/Tutorial.ipynb
@@ -39,7 +39,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "test = torch.load('test.pt')"
+    "test = torch.load(\"test.pt\")"
    ]
   },
   {
@@ -53,7 +53,7 @@
    "source": [
     "tmp = !ls | grep pt\n",
     "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
-    "model = torch.load('sample.pt', map_location=device)"
+    "model = torch.load(\"sample.pt\", map_location=device)"
    ]
   },
   {

diff --git a/model_training.ipynb b/model_training.ipynb
@@ -41,7 +41,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "data = torch.load('train.pt')"
+    "data = torch.load(\"train.pt\")"
    ]
   },
   {