diff --git a/doc/design/gan_api.md b/doc/design/gan_api.md index 5764112f3c84512c30aacfbf412f7660702a077f..fb41df8615f73d9fd4c32995eab265833eac1a55 100644 --- a/doc/design/gan_api.md +++ b/doc/design/gan_api.md @@ -7,7 +7,7 @@ It applies several important concepts in machine learning system design, includi In our GAN design, we wrap it as a user-friendly easily customized python API to design different models. We take the conditional DC-GAN (Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks [https://arxiv.org/abs/1511.06434]) as an example due to its good performance on image generation.

-
+
Figure 1. The overall running logic of GAN. The black solid arrows indicate the forward pass; the green dashed arrows indicate the backward pass of generator training; the red dashed arrows indicate the backward pass of the discriminator training. The BP pass of the green (red) arrow should only update the parameters in the green (red) boxes. The diamonds indicate the data providers. d\_loss and g\_loss marked in red and green are the two targets we would like to run.