Spaces:
Sleeping
Sleeping
Muhammad Naufal Rizqullah
commited on
Commit
·
439c972
1
Parent(s):
eb42124
change readme and change weight model
Browse files
README.md
CHANGED
@@ -1,49 +1,13 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
* **Model architecture and hyperparameters optimized**: for Kaggle environment with 15GB VRAM
|
15 |
-
* **Embedding used instead of one-hot encoding**: for training labels to avoid using 0 on labels
|
16 |
-
* **Implemented using PyTorch Lightning framework**
|
17 |
-
|
18 |
-
## Training Details
|
19 |
-
---------------
|
20 |
-
|
21 |
-
### Training Epochs
|
22 |
-
----------------
|
23 |
-
|
24 |
-
* Approximately 1000 epochs
|
25 |
-
|
26 |
-
### Model Architecture Compromises
|
27 |
-
-----------------------------
|
28 |
-
|
29 |
-
* **Reduced Size of latent space (z dim) and Embedding**: due to VRAM limitations
|
30 |
-
* **Limited features for generator and critic networks**: due to VRAM limitations
|
31 |
-
* **Image size limitations**: due to VRAM limitations
|
32 |
-
* **Dataset used**: [Shoe vs Sandal vs Boot Image Dataset (15K Images)](https://www.kaggle.com/datasets/hasibalmuzdadid/shoe-vs-sandal-vs-boot-dataset-15k-images)
|
33 |
-
|
34 |
-
## Results
|
35 |
-
---------
|
36 |
-
|
37 |
-
Despite the architectural compromises, the model produces reasonable results. However, the quality of the generated images may not be optimal due to the limited dataset and VRAM constraints.
|
38 |
-
|
39 |
-
## Demo
|
40 |
-
-------------
|
41 |
-
The demo of this project is deployed on Hugging Face's model hub and uses the Gradio framework to provide a user-friendly interface for interacting with the model. You can try out the demo by visiting [this link](https://huggingface.co/spaces/SkylarWhite/57894).
|
42 |
-
|
43 |
-
|
44 |
-
## Future Work
|
45 |
-
-------------
|
46 |
-
|
47 |
-
* **Experiment with larger datasets and more complex model architectures**
|
48 |
-
* **Investigate alternative optimization techniques to improve model performance**
|
49 |
-
* **Explore other applications of Conditional GANs in computer vision**
|
|
|
1 |
+
---
|
2 |
+
title: '57894'
|
3 |
+
emoji: 🐠
|
4 |
+
colorFrom: indigo
|
5 |
+
colorTo: purple
|
6 |
+
sdk: gradio
|
7 |
+
sdk_version: 4.41.0
|
8 |
+
app_file: app.py
|
9 |
+
pinned: false
|
10 |
+
license: mit
|
11 |
+
---
|
12 |
+
|
13 |
+
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
weights/{epoch=999-step=96000.ckpt → epoch=957-step=1164300.ckpt}
RENAMED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1602e9cb80fcfbc39ed2c5c74873744a546cbab346730aaa0f25b02140bc1d2f
|
3 |
+
size 157799852
|