Error in the The operator 'aten::_linalg_solve_ex.result' in the aug_…

…transforms function in the Data Augmentation Fixes fastai#588
Akhil-Raj · Jun 5, 2023 · ac14ac7 · ac14ac7
1 parent f6e40a8
commit ac14ac7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/02_production.ipynb b/02_production.ipynb
@@ -926,7 +926,8 @@
     }
    ],
    "source": [
-    "bears = bears.new(item_tfms=Resize(128), batch_tfms=aug_transforms(mult=2))\n",
+    "custom_aug_transforms = [RandomResizedCrop(128, min_scale=0.35), Flip(), Brightness(), Contrast(), Rotate(max_deg=10.0)]\n",
+    "bears = bears.new(item_tfms=Resize(128), batch_tfms=custom_aug_transforms)\n",
     "dls = bears.dataloaders(path)\n",
     "dls.train.show_batch(max_n=8, nrows=2, unique=True)"
    ]