update: yolo-nas colab

wandb · Aug 14, 2023 · 400298d · 400298d
1 parent 45cea3b
commit 400298d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/colabs/README.md b/colabs/README.md
@@ -24,6 +24,7 @@
 | 🧨 Dreambooth-Keras Train | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](http://wandb.me/db-keras-train) |
 | 🧨 Dreambooth-Keras Inference | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](http://wandb.me/db-keras-inference) |
 | Kaolin-Wisp | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](http://wandb.me/vqad-colab) |
+| Super Gradients | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](http://wandb.me/yolo-nas-colab) |
 
 # 🏋🏽‍♂️ W&B Features
 

diff --git a/colabs/super-gradients/assets/val-table.gif b/colabs/super-gradients/assets/val-table.gif
diff --git a/colabs/super-gradients/yolo_nas.ipynb b/colabs/super-gradients/yolo_nas.ipynb
@@ -348,6 +348,15 @@
    "source": [
     "trainer.train(model=net, training_params=train_params, train_loader=train_data, valid_loader=val_data)"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Here's how the validation predictions look at the end of the training 👇\n",
+    "\n",
+    "![](./assets/val-table.gif)"
+   ]
   }
  ],
  "metadata": {