Skip to content

Instantly share code, notes, and snippets.

@George3d6
Created August 14, 2018 17:32
Show Gist options
  • Save George3d6/e67c4853481267731a4d19ecdc6bbf37 to your computer and use it in GitHub Desktop.
Save George3d6/e67c4853481267731a4d19ecdc6bbf37 to your computer and use it in GitHub Desktop.
global const 𝜀=Float32(1e-8) # a small number prevent from getting NaN in logs
𝑱d(𝗪d,x,Gz) = -mean(log.(D(𝗪d,x)+𝜀)+log.(1-D(𝗪d,Gz)+𝜀))/2 # Discriminator Loss
𝑱g(𝗪g, 𝗪d, z) = -mean(log.(D(𝗪d,G(𝗪g,z))+𝜀)) # Generator Loss
∇d = grad(𝑱d) # Discriminator gradient
∇g = grad(𝑱g) # Generator gradient
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment