From e70a439671a052bbcf79639dcf290f7bbd715cd2 Mon Sep 17 00:00:00 2001
From: Mark Kurtz <mark@neuralmagic.com>
Date: Thu, 29 Jul 2021 10:40:55 -0400
Subject: [PATCH 1/2] YOLOv5 Sparse Transfer Learning Command Fixes (#342)

---
 .../tutorials/yolov5_sparse_transfer_learning.md          | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/integrations/ultralytics-yolov5/tutorials/yolov5_sparse_transfer_learning.md b/integrations/ultralytics-yolov5/tutorials/yolov5_sparse_transfer_learning.md
index dae5d79508a..0d1502a292e 100644
--- a/integrations/ultralytics-yolov5/tutorials/yolov5_sparse_transfer_learning.md
+++ b/integrations/ultralytics-yolov5/tutorials/yolov5_sparse_transfer_learning.md
@@ -117,11 +117,11 @@ The recipes are specific to the sparsification type, so the training command wil
 1) Select the proper command to run based on the model and the sparsification type of the model you chose earlier.
    - YOLOv5s Pruned transfer learning:
      ```bash
-     python train.py --data voc.yaml --cfg ../models/yolov5s.yaml --weights zoo:cv/detection/yolov5-s/pytorch/ultralytics/coco/pruned-aggressive_96 --hyp data/hyp.finetune.yaml --recipe ../recipes/yolov5s.transfer_learn_pruned.md
+     python train.py --data voc.yaml --cfg ../models/yolov5s.yaml --weights zoo:cv/detection/yolov5-s/pytorch/ultralytics/coco/pruned-aggressive_96?recipe_type=transfer --hyp data/hyp.finetune.yaml --recipe ../recipes/yolov5.transfer_learn_pruned.md
      ```
    - YOLOv5s Pruned-Quantized transfer learning:
      ```bash
-     python train.py --data voc.yaml --cfg ../models/yolov5s.yaml --weights zoo:cv/detection/yolov5-s/pytorch/ultralytics/coco/pruned_quant-aggressive_94 --hyp data/hyp.finetune.yaml --recipe ../recipes/yolov5s.transfer_learn_pruned_quantized.md
+     python train.py --data voc.yaml --cfg ../models/yolov5s.yaml --weights zoo:cv/detection/yolov5-s/pytorch/ultralytics/coco/pruned_quant-aggressive_94?recipe_type=transfer --hyp data/hyp.finetune.yaml --recipe ../recipes/yolov5.transfer_learn_pruned_quantized.md
      ```
    - YOLOv5s Baseline transfer learning:
      ```bash
@@ -129,11 +129,11 @@ The recipes are specific to the sparsification type, so the training command wil
      ```
    - YOLOv5l Pruned transfer learning:
      ```bash
-     python train.py --data voc.yaml --cfg ../models/yolov5l.yaml --weights zoo:cv/detection/yolov5-l/pytorch/ultralytics/coco/pruned-aggressive_98 --hyp data/hyp.finetune.yaml --recipe ../recipes/yolov5l.transfer_learn_pruned.md
+     python train.py --data voc.yaml --cfg ../models/yolov5l.yaml --weights zoo:cv/detection/yolov5-l/pytorch/ultralytics/coco/pruned-aggressive_98?recipe_type=transfer --hyp data/hyp.finetune.yaml --recipe ../recipes/yolov5.transfer_learn_pruned.md
      ```
    - YOLOv5l Pruned-Quantized transfer learning:
      ```bash
-     python train.py --data voc.yaml --cfg ../models/yolov5l.yaml --weights zoo:cv/detection/yolov5-l/pytorch/ultralytics/coco/pruned_quant-aggressive_95 --hyp data/hyp.finetune.yaml --recipe ../recipes/yolov5l.transfer_learn_pruned_quantized.md
+     python train.py --data voc.yaml --cfg ../models/yolov5l.yaml --weights zoo:cv/detection/yolov5-l/pytorch/ultralytics/coco/pruned_quant-aggressive_95?recipe_type=transfer --hyp data/hyp.finetune.yaml --recipe ../recipes/yolov5.transfer_learn_pruned_quantized.md
      ```
    - YOLOv5l Baseline transfer learning:
      ```bash

From 05326cdaa1a12ee43f2383b31a07731869d3055b Mon Sep 17 00:00:00 2001
From: Mark Kurtz <mark@neuralmagic.com>
Date: Thu, 29 Jul 2021 11:04:44 -0400
Subject: [PATCH 2/2] Update README Highlights for yolo and bert models (#344)

---
 README.md | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/README.md b/README.md
index a92f523ff55..d40d4470614 100644
--- a/README.md
+++ b/README.md
@@ -91,6 +91,12 @@ The [GitHub repository](https://github.com/neuralmagic/sparseml) contains integr
     <a href="https://github.com/neuralmagic/sparseml/tree/main/integrations/ultralytics-yolov3/tutorials/sparsifying_yolov3_using_recipes.md">
         <img src="https://docs.neuralmagic.com/docs/source/tutorials/detection_yolov3.png" width="136px" />
     </a>
+    <a href="https://github.com/neuralmagic/sparseml/tree/main/integrations/ultralytics-yolov5/tutorials/sparsifying_yolov5_using_recipes.md">
+        <img src="https://docs.neuralmagic.com/docs/source/tutorials/detection_yolov5.png" width="136px" />
+    </a>
+    <a href="https://github.com/neuralmagic/sparseml/tree/main/integrations/huggingface-transformers/tutorials/sparsifying_bert_using_recipes.md">
+        <img src="https://docs.neuralmagic.com/docs/source/tutorials/nlp_bert.png" width="136px" />
+    </a>
 </p>
 
 ### Transfer Learning from Sparse Models
@@ -99,6 +105,12 @@ The [GitHub repository](https://github.com/neuralmagic/sparseml) contains integr
     <a href="https://github.com/neuralmagic/sparseml/tree/main/integrations/pytorch/notebooks/sparse_quantized_transfer_learning.ipynb">
         <img src="https://docs.neuralmagic.com/docs/source/tutorials/classification_resnet-50.png" width="136px" />
     </a>
+    <a href="https://github.com/neuralmagic/sparseml/blob/main/integrations/ultralytics-yolov3/tutorials/yolov3_sparse_transfer_learning.md">
+        <img src="https://docs.neuralmagic.com/docs/source/tutorials/detection_yolov3.png" width="136px" />
+    </a>
+    <a href="https://github.com/neuralmagic/sparseml/blob/main/integrations/ultralytics-yolov5/tutorials/yolov5_sparse_transfer_learning.md">
+        <img src="https://docs.neuralmagic.com/docs/source/tutorials/detection_yolov5.png" width="136px" />
+    </a>
 </p>