diff --git a/training/train_instance_segmentation_model.ipynb b/training/train_instance_segmentation_model.ipynb
index cd49153..d4249af 100644
--- a/training/train_instance_segmentation_model.ipynb
+++ b/training/train_instance_segmentation_model.ipynb
@@ -409,7 +409,7 @@
     "In this section, we can set up everything connected to the actual training. You can change preprocessing, batch size, epochs, add callbacks, augmentations, change optimizers, schedulers, and more. Please refer to the [complete documentation](https://github.com/luxonis/luxonis-train/tree/main/configs). \n",
     "\n",
     "#### Augmentations\n",
-    "In this tutorial, we'll leave most things as they are; the only change will be adding some augmentations. `Luxonis-train` uses [`Albumentations`](https://albumentations.ai/) for augmentations by adding custom ones like `Mosaic4` and `MixUp`. You can use [this demo](https://demo.albumentations.ai/) to experiment and find those that work for your specific training run. In this training run, we'll add `Affine`, `HorizontalFlip`, `ColorJitter`, `Sharpen` and `Mosaic4` augmentations, but feel free to edit this.\n",
+    "In this tutorial, we'll leave most things as they are; the only change will be adding some augmentations. `Luxonis-train` uses [`Albumentations`](https://albumentations.ai/) for augmentations by adding custom ones like `MixUp`. You can use [this demo](https://demo.albumentations.ai/) to experiment and find those that work for your specific training run. In this training run, we'll add `HorizontalFlip`, `ColorJitter` and `Sharpen` augmentations, but feel free to edit this.\n",
     "\n",
     "#### Callbacks \n",
     "Callbacks are very helpful when merging more functionalities into a single training run. For example, we want to train the model, evaluate it on a test subset, export it, and create an archive. These steps can be defined through the config and done by a single call. You can check out all the available callbacks [here](https://github.com/luxonis/luxonis-train/tree/main/luxonis_train/callbacks)."
@@ -456,7 +456,7 @@
     "\n",
     "trainer:\n",
     "  preprocessing:\n",
-    "    train_image_size: [640, 640]\n",
+    "    train_image_size: [448, 448]\n",
     "    keep_aspect_ratio: true\n",
     "    normalize:\n",
     "      active: true\n",
@@ -464,12 +464,6 @@
     "        mean: [0., 0., 0.]\n",
     "        std: [1, 1, 1]\n",
     "    augmentations:\n",
-    "      - name: Affine\n",
-    "        params:\n",
-    "          scale: [0.7, 1.6]\n",
-    "          rotate: 30\n",
-    "          shear: 10\n",
-    "          p: 0.3\n",
     "      - name: HorizontalFlip\n",
     "        params:\n",
     "          p: 0.3\n",
@@ -484,10 +478,6 @@
     "        params:\n",
     "          p: 0.3\n",
     "      - name: RandomRotate90\n",
-    "      - name: Mosaic4\n",
-    "        params:\n",
-    "          out_width: 640\n",
-    "          out_height: 640\n",
     "\n",
     "\n",
     "  batch_size: 8\n",
@@ -516,7 +506,7 @@
     "      warmup_epochs: 5\n",
     "      warmup_bias_lr: 0.0\n",
     "      warmup_momentum: 0.8\n",
-    "      lr: 0.005\n",
+    "      lr: 0.01\n",
     "      lre: 0.0001\n",
     "      momentum: 0.937     \n",
     "      weight_decay: 0.001\n",