Skip to content

Commit 1493bd6

Browse files
Merge pull request #504 from ilopezfr:patch-1
PiperOrigin-RevId: 245130358
2 parents 5e34bd1 + 1344a17 commit 1493bd6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

site/en/r2/tutorials/generative/style_transfer.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@
217217
"def load_img(path_to_img):\n",
218218
" max_dim = 512\n",
219219
" img = tf.io.read_file(path_to_img)\n",
220-
" img = tf.image.decode_jpeg(img)\n",
220+
" img = tf.image.decode_image(img, channels=3)\n",
221221
" img = tf.image.convert_image_dtype(img, tf.float32)\n",
222222
"\n",
223223
" shape = tf.cast(tf.shape(img)[:-1], tf.float32)\n",

0 commit comments

Comments
 (0)