|
@@ -182,7 +182,7 @@
|
|
" random_seed=None, keep_pct=keep_pct, num_workers=num_works)\n",
|
|
" random_seed=None, keep_pct=keep_pct, num_workers=num_works)\n",
|
|
"\n",
|
|
"\n",
|
|
"def get_crit_data(classes, bs, sz):\n",
|
|
"def get_crit_data(classes, bs, sz):\n",
|
|
- " src = ImageList.from_folder(path, include=classes, recurse=True).random_split_by_pct(0.1, seed=42)\n",
|
|
|
|
|
|
+ " src = ImageList.from_folder(path, include=classes, recurse=True).split_by_rand_pct(0.1, seed=42)\n",
|
|
" ll = src.label_from_folder(classes=classes)\n",
|
|
" ll = src.label_from_folder(classes=classes)\n",
|
|
" data = (ll.transform(get_transforms(max_zoom=2.), size=sz)\n",
|
|
" data = (ll.transform(get_transforms(max_zoom=2.), size=sz)\n",
|
|
" .databunch(bs=bs).normalize(imagenet_stats))\n",
|
|
" .databunch(bs=bs).normalize(imagenet_stats))\n",
|