locally everything works

pyt-team · Nov 7, 2023 · 6621531 · 6621531
1 parent 44e75ed
commit 6621531
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/hypergraph/allset_transformer_train.ipynb b/tutorials/hypergraph/allset_transformer_train.ipynb
@@ -502,7 +502,7 @@
    "source": [
     "torch.manual_seed(0)\n",
     "test_interval = 5\n",
-    "num_epochs = 30\n",
+    "num_epochs = 10\n",
     "\n",
     "epoch_loss = []\n",
     "for epoch_i in range(1, num_epochs + 1):\n",