agjn7.jl

#original code: https://github.com/ekinakyurek/GAN-70-Lines-of-Julia/blob/66a60a6ea4532841ee647f08759ae9b1ace0c892/gan.jl#L42
function train_model(𝗪, data, 𝞗, optim)
    gloss=dloss=counter=0.0;
    B =
    for generation=1:𝞗[:epochs]
        for n=1:32:(length(data[1,1,1,:]) - 33)
            x = data[:,:,:,n:n+31]

            Gz = G(𝗪[1],𝒩(𝞗[:ginp],𝞗[:batchsize]))
            update!(𝗪[2], ∇d(𝗪[2],x,Gz), optim[2])

            z=𝒩(𝞗[:ginp],2*𝞗[:batchsize])
            update!(𝗪[1], ∇g(𝗪[1], 𝗪[2], z), optim[1])
        end
        # Compute the total losses, log them and save some images
        log_model(𝗪, data, 𝞗, generation)
    end
end