diff --git a/doc/design/gan_api.md b/doc/design/gan_api.md index 1a7d0df116..a1626e50d7 100644 --- a/doc/design/gan_api.md +++ b/doc/design/gan_api.md @@ -27,10 +27,15 @@ In our GAN design, we wrap it as a user-friendly easily customized python API to | Concat op (done) | ? | N (Cond) | | Repmat op (done) | ? | N (Cond) | +
+
+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.
+
-Borrow this photo from the original DC-GAN paper.
+Photo borrowed from the original DC-GAN paper.