Cleaned up formatting of nbody.ipynb

arnab39 · Mar 14, 2024 · ca63da7 · ca63da7
1 parent 0cb6244
commit ca63da7
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/tutorials/nbody/nbody.ipynb b/tutorials/nbody/nbody.ipynb
@@ -14,8 +14,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import warnings\n",
-    "warnings.filterwarnings('ignore')\n",
     "import torch\n",
     "from tqdm import tqdm\n",
     "\n",
@@ -375,7 +373,6 @@
    "source": [
     "epochs = 20\n",
     "\n",
-    "# finetuning the prediction network with the canonicalizer for STL10 dataset\n",
     "for epoch in range(epochs):\n",
     "    tqdm_bar = tqdm(enumerate(train_loader), desc=f\"Epoch {epoch}\", total=len(train_loader))\n",
     "                 \n",