Created using Colaboratory

Lechtr · Apr 12, 2021 · cac8a76 · cac8a76
1 parent 2eab46e
commit cac8a76
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/tutorial.ipynb b/tutorial.ipynb
@@ -550,7 +550,7 @@
         "colab": {
           "base_uri": "https://localhost:8080/"
         },
-        "outputId": "4576b05f-d6d1-404a-fc99-5663c71e3dc4"
+        "outputId": "9b022435-4197-41fc-abea-81f86ce857d0"
       },
       "source": [
         "!git clone https://github.com/ultralytics/yolov5  # clone repo\n",
@@ -561,14 +561,14 @@
         "from IPython.display import Image, clear_output  # to display images\n",
         "\n",
         "clear_output()\n",
-        "print('Setup complete. Using torch %s %s' % (torch.__version__, torch.cuda.get_device_properties(0) if torch.cuda.is_available() else 'CPU'))"
+        "print(f\"Setup complete. Using torch {torch.__version__} ({torch.cuda.get_device_properties(0).name if torch.cuda.is_available() else 'CPU'})\")"
       ],
-      "execution_count": 1,
+      "execution_count": 31,
       "outputs": [
         {
           "output_type": "stream",
           "text": [
-            "Setup complete. Using torch 1.8.1+cu101 _CudaDeviceProperties(name='Tesla V100-SXM2-16GB', major=7, minor=0, total_memory=16160MB, multi_processor_count=80)\n"
+            "Setup complete. Using torch 1.8.1+cu101 (Tesla V100-SXM2-16GB)\n"
           ],
           "name": "stdout"
         }
@@ -681,7 +681,7 @@
         "torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017val.zip', 'tmp.zip')\n",
         "!unzip -q tmp.zip -d ../ && rm tmp.zip"
       ],
-      "execution_count": 2,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "display_data",
@@ -721,7 +721,7 @@
         "# Run YOLOv5x on COCO val2017\n",
         "!python test.py --weights yolov5x.pt --data coco.yaml --img 640 --iou 0.65"
       ],
-      "execution_count": 6,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -922,7 +922,7 @@
         "# Train YOLOv5s on COCO128 for 3 epochs\n",
         "!python train.py --img 640 --batch 16 --epochs 3 --data coco128.yaml --weights yolov5s.pt --nosave --cache"
       ],
-      "execution_count": 12,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -1259,4 +1259,4 @@
       "outputs": []
     }
   ]
-}
+}