|
@@ -111,7 +111,7 @@
|
|
|
" o.save(path_gen/names[i].name)\n",
|
|
|
" i += 1\n",
|
|
|
" \n",
|
|
|
- "def save_gen_images(learn_gen):\n",
|
|
|
+ "def save_gen_images():\n",
|
|
|
" if path_gen.exists(): shutil.rmtree(path_gen)\n",
|
|
|
" path_gen.mkdir(exist_ok=True)\n",
|
|
|
" data_gen = get_data(bs=bs, sz=sz, keep_pct=0.085)\n",
|
|
@@ -418,7 +418,7 @@
|
|
|
"metadata": {},
|
|
|
"outputs": [],
|
|
|
"source": [
|
|
|
- "save_gen_images(gen_name)"
|
|
|
+ "save_gen_images()"
|
|
|
]
|
|
|
},
|
|
|
{
|