opensource/DeOldify
3
0
Fork 0
mirror of https://github.com/jantic/DeOldify.git synced 2026-08-30 18:02:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #188 from asears/master

Browse Source 
fix warning random_split_by_pct` is deprecated, please use `split_by_rand_pct'
This commit is contained in:
Jason Antic
2020-02-05 19:30:12 -08:00
committed by GitHub
parent f8fe110866 46a499fb62
commit b1eb7c5d67
5 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ColorizeTrainingArtistic.ipynb
+1 -1
View File
@@ -89,7 +89,7 @@
" random_seed=None, keep_pct=keep_pct)\n",
"\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",
" data = (ll.transform(get_transforms(max_zoom=2.), size=sz)\n",
" .databunch(bs=bs).normalize(imagenet_stats))\n",
ColorizeTrainingStable.ipynb
+1 -1
View File
@@ -89,7 +89,7 @@
" random_seed=None, keep_pct=keep_pct)\n",
"\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",
" data = (ll.transform(get_transforms(max_zoom=2.), size=sz)\n",
" .databunch(bs=bs).normalize(imagenet_stats))\n",
ColorizeTrainingStableLargeBatch.ipynb
+1 -1
View File
@@ -182,7 +182,7 @@
" random_seed=None, keep_pct=keep_pct, num_workers=num_works)\n",
"\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",
" data = (ll.transform(get_transforms(max_zoom=2.), size=sz)\n",
" .databunch(bs=bs).normalize(imagenet_stats))\n",
ColorizeTrainingVideo.ipynb
+1 -1
View File
@@ -89,7 +89,7 @@
" random_seed=None, keep_pct=keep_pct, xtra_tfms=xtra_tfms)\n",
"\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",
" data = (ll.transform(get_transforms(max_zoom=2.), size=sz)\n",
" .databunch(bs=bs).normalize(imagenet_stats))\n",
ColorizeTrainingWandb.ipynb
+1 -1
View File
@@ -203,7 +203,7 @@
" return _f\n",
"\n",
"def get_crit_data(classes, bs, sz, pct=1.0):\n",
" src = ImageList.from_folder(path, include=classes, recurse=True).filter_by_func(filter_data(pct)).random_split_by_pct(0.1)\n",
" src = ImageList.from_folder(path, include=classes, recurse=True).filter_by_func(filter_data(pct)).split_by_rand_pct(0.1)\n",
" ll = src.label_from_folder(classes=classes)\n",
" data = (ll.transform(get_transforms(max_zoom=2.), size=sz)\n",
" .databunch(bs=bs).normalize(imagenet_stats))\n",