Fix typos in tensorboard example

examples/tensorboard/tensorboard_profiler_tutorial.ipynb

@@ -147,12 +147,13 @@
     "     T.ToTensor(),\n",
     "     T.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])\n",
     "train_set = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform)\n",
-    "# train_loader = torch.utils.data.DataLoader(train_set, batch_size=32, shuffle=True, num_workers=1)\n",
+    "train_loader = torch.utils.data.DataLoader(train_set, batch_size=32, shuffle=True, num_workers=1)\n",
     "\n",
     "# This requires more than 16GB of GPU memory\n",
     "# train_loader = torch.utils.data.DataLoader(train_set, batch_size=32, shuffle=True, num_workers=2)\n",
     "\n",
-    "train_loader = torch.utils.data.DataLoader(train_set, batch_size=16, shuffle=True, num_workers=2)"
+    "# This works on the V100 gpu and provides faster training\n",
+    "# train_loader = torch.utils.data.DataLoader(train_set, batch_size=16, shuffle=True, num_workers=2)"
    ]
   },
   {
@@ -254,7 +255,7 @@
     "-   `on_trace_ready` - callable that is called at the end of each cycle;\n",
     "    In this example we use `torch.profiler.tensorboard_trace_handler` to\n",
     "    generate result files for TensorBoard. After profiling, result files\n",
-    "    will be saved into the `./log/resnet18` directory. Specify this\n",
+    "    will be saved into the `./log/efficientnet_b0` directory. Specify this\n",
     "    directory as a `logdir` parameter to analyze profile in TensorBoard.\n",
     "-   `record_shapes` - whether to record shapes of the operator inputs.\n",
     "-   `profile_memory` - Track tensor memory allocation/deallocation.\n",
@@ -484,7 +485,7 @@
     "\n",
     "In this example, we follow the \\\"Performance Recommendation\\\" and set\n",
     "`num_workers` as below, pass a different name such as\n",
-    "`./log/resnet18_4workers` to `tensorboard_trace_handler`, and run it\n",
+    "`./log/efficientnet_b0_4workers` to `tensorboard_trace_handler`, and run it\n",
     "again.\n",
     "\n",
     "``` {.sourceCode .}\n",
@@ -706,7 +707,7 @@
     "displays nothing. You can work around by typing `chrome://tracing` in\n",
     "your Chrome Browser.\n",
     "\n",
-    "\\- Copy the `trace.json` file under `~/profiler_tutorial/log/resnet18`\n",
+    "\\- Copy the `trace.json` file under `~/profiler_tutorial/log/efficientnet_b0`\n",
     "directory to the Windows. You may need to copy the file by using `scp`\n",
     "if the file is located in a remote location.\n",
     "\n",