Gholamreza commited on
Commit
05bece3
·
verified ·
1 Parent(s): 8895b4f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -26
README.md CHANGED
@@ -1,26 +1,14 @@
1
- # Generating MNIST digits using Conditional GAN
2
-
3
- This is a simple implementation of Conditional Generative Adversarial Networks (GAN) for generating MNIST digits.
4
-
5
- ![cover](demos/gen_all_digits.png)
6
-
7
- I use simple BCE loss function for calculating the loss and Adam optimizer (lr=0.0001) for training.
8
-
9
- ## Architecture
10
-
11
- - The **generator** is series of Linear layers with BatchNorm and ReLU activations.
12
- - The **discriminator** is a series of Linear layers with BatchNorm andLeakyReLU activations.
13
- - The Conditioning class is appended to the noise vector as a one-hot vector.
14
-
15
- ## Huggingface Space
16
-
17
- You can try generating digits using this model on Huggingface Space.
18
- https://huggingface.co/spaces/gholamreza/Conditional-GAN-MNIST
19
-
20
- ![Huggingface Space](demos/gradio_app.png)
21
-
22
- ## Training History
23
-
24
- ![losses_plot](demos/losses.png)
25
-
26
- Visit https://github.com/gholamrezadar/GAN-MNIST for a simpler version of this code and more details.
 
1
+ ---
2
+ title: Conditional GAN MNIST
3
+ emoji: 💻
4
+ colorFrom: purple
5
+ colorTo: green
6
+ sdk: gradio
7
+ sdk_version: 5.7.1
8
+ app_file: app.py
9
+ pinned: false
10
+ license: apache-2.0
11
+ short_description: This is a simple implementation of Conditional Generative Ad
12
+ ---
13
+
14
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference