more parameters explained

example.ipynb

@@ -110,7 +110,12 @@
     "import pytorch_lightning as pl\n",
     "\n",
     "\n",
-    "plmodel = PLModel(cfg, backbone, draw=True)\n",
+    "plmodel = PLModel(\n",
+    "    cfg, \n",
+    "    backbone, \n",
+    "    draw=True,  # draw on each epoch end\n",
+    "    cached=True,  # cache the features into cpu memory in first epoch\n",
+    ")\n",
     "# plmodel.validation_epoch_end() is called on validation epoch to draw\n",
     "plmodel = plmodel.cuda()"
    ]
@@ -538,7 +543,7 @@
     "    accelerator=\"gpu\",\n",
     "    devices=[0],\n",
     "    gradient_clip_val=0.5,\n",
-    "    precision=16,\n",
+    "    precision=16,  # set to 32 if you use GPU before 2080\n",
     "    limit_train_batches=1.0,\n",
     "    limit_val_batches=1.0,\n",
     "    enable_checkpointing=False,\n",