Skip to content

Commit

Permalink
change random.normal to random.uniform
Browse files Browse the repository at this point in the history
  • Loading branch information
sachinprasadhs committed Aug 3, 2023
1 parent 6ed3e11 commit 68ad4d8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/generative/ipynb/wgan_gp.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"\n",
"**Author:** [A_K_Nain](https://twitter.com/A_K_Nain)<br>\n",
"**Date created:** 2020/05/9<br>\n",
"**Last modified:** 2020/05/9<br>\n",
"**Last modified:** 2023/08/3<br>\n",
"**Description:** Implementation of Wasserstein GAN with Gradient Penalty."
]
},
Expand Down Expand Up @@ -351,7 +351,7 @@
" and added to the discriminator loss.\n",
" \"\"\"\n",
" # Get the interpolated image\n",
" alpha = tf.random.normal([batch_size, 1, 1, 1], 0.0, 1.0)\n",
" alpha = tf.random.uniform([batch_size, 1, 1, 1], 0.0, 1.0)\n",
" diff = fake_images - real_images\n",
" interpolated = real_images + alpha * diff\n",
"\n",
Expand Down

0 comments on commit 68ad4d8

Please sign in to comment.