From 90a7a93e5d13105ba111fe1fae46148aa077c49e Mon Sep 17 00:00:00 2001 From: Akhil Raj Date: Mon, 5 Jun 2023 22:12:16 +0530 Subject: [PATCH] Error in the The operator 'aten::_linalg_solve_ex.result' in the aug_transforms function in the Data Augmentation Fixes #588 --- 02_production.ipynb | 30 ++++++++++++++++++++++++++++-- 1 file changed, 28 insertions(+), 2 deletions(-) diff --git a/02_production.ipynb b/02_production.ipynb index 245f46988..779619c0f 100644 --- 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)" ] @@ -2004,9 +2005,34 @@ "split_at_heading": true }, "kernelspec": { - "display_name": "Python 3", + "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" + }, + "language_info": { + "codemirror_mode": { + "name": "ipython", + "version": 3 + }, + "file_extension": ".py", + "mimetype": "text/x-python", + "name": "python", + "nbconvert_exporter": "python", + "pygments_lexer": "ipython3", + "version": "3.9.13" + }, + "toc": { + "base_numbering": 1, + "nav_menu": {}, + "number_sections": true, + "sideBar": true, + "skip_h1_title": false, + "title_cell": "Table of Contents", + "title_sidebar": "Contents", + "toc_cell": false, + "toc_position": {}, + "toc_section_display": true, + "toc_window_display": false } }, "nbformat": 4,