Merge pull request #3 from Barabaika/main

issue2
AIRI-Institute · Feb 13, 2024 · 4b5995e · 4b5995e
2 parents 4d2a9ef + 0cab94a
commit 4b5995e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/SEMA_1D/SEMA-1D_inference.ipynb b/SEMA_1D/SEMA-1D_inference.ipynb
@@ -243,11 +243,11 @@
     "    model.eval()\n",
     "    model.cuda()\n",
     "\n",
-    "with torch.no_grad():\n",
-    "    preds=[]\n",
-    "    for it in tqdm(test_ds):\n",
-    "        preds.append(model.forward(it['token_ids'].cuda())[0][0][:,1].cpu().numpy())\n",
-    "res.append(preds)"
+    "    with torch.no_grad():\n",
+    "        preds=[]\n",
+    "        for it in tqdm(test_ds):\n",
+    "            preds.append(model.forward(it['token_ids'].cuda())[0][0][:,1].cpu().numpy())\n",
+    "    res.append(preds)"
    ]
   },
   {