chris/fastbook
1
0
Fork 0
mirror of https://github.com/fastai/fastbook.git synced 2025-04-05 02:10:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Error in the The operator 'aten::_linalg_solve_ex.result' in the aug_transforms function in the Data Augmentation

Browse Source 
Fixes #588
This commit is contained in:
Akhil Raj 2023-06-05 22:20:14 +05:30
parent f6e40a805d
commit ac14ac7bb9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
02_production.ipynb
View File

@ -926,7 +926,8 @@
} }
], ],
"source": [ "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 = bears.dataloaders(path)\n",
"dls.train.show_batch(max_n=8, nrows=2, unique=True)" "dls.train.show_batch(max_n=8, nrows=2, unique=True)"
] ]