Update README.md
Browse files
README.md
CHANGED
|
@@ -32,6 +32,9 @@ Dataset was scraped from pinterest and cropped using dlib. Further symmetry and
|
|
| 32 |
|
| 33 |
Training was done locally using StyleGAN3 with an RTX 4090 and cuda 11.8. Training took roughly 100 hours but hyperparameter tuning/restarting was modified (many times) about 3/4 of the way through.
|
| 34 |
|
|
|
|
|
|
|
|
|
|
| 35 |
- Configuration: `stylegan3-t`
|
| 36 |
- GPUs: `1`
|
| 37 |
- Batch Size: `32`
|
|
@@ -40,6 +43,15 @@ Training was done locally using StyleGAN3 with an RTX 4090 and cuda 11.8. Traini
|
|
| 40 |
- Image Resolution: '256x256'
|
| 41 |
- Final fid50k_full value (this pickle): `10.161478712247181`
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
#### Differences from v1.0.0 (10 months in the making)
|
| 44 |
- Noticeable improvements in quality and detail
|
| 45 |
- More variety in hairstyles, hair colors, eye colors, and even accessories (earrings and sunglasses)
|
|
|
|
| 32 |
|
| 33 |
Training was done locally using StyleGAN3 with an RTX 4090 and cuda 11.8. Training took roughly 100 hours but hyperparameter tuning/restarting was modified (many times) about 3/4 of the way through.
|
| 34 |
|
| 35 |
+
You have two versions to choose from, depending on your preference.
|
| 36 |
+
|
| 37 |
+
#### Main Version (v2.0.0)
|
| 38 |
- Configuration: `stylegan3-t`
|
| 39 |
- GPUs: `1`
|
| 40 |
- Batch Size: `32`
|
|
|
|
| 43 |
- Image Resolution: '256x256'
|
| 44 |
- Final fid50k_full value (this pickle): `10.161478712247181`
|
| 45 |
|
| 46 |
+
#### Alternate Version (v2.1.0_alt)
|
| 47 |
+
- Dataset was reduced to 60% (better quality)
|
| 48 |
+
- Trained for another 2000 kimg
|
| 49 |
+
- Final fid50k_full value (this pickle): `5.079479576662786`
|
| 50 |
+
|
| 51 |
+
#### Differences between main and alternate version
|
| 52 |
+
- Main Version: Better variety and realism, but higher chance of generating bad "outlier" faces
|
| 53 |
+
- Alternate Version: Better Overall Quality, less variety in faces (less chances of generating bad faces)
|
| 54 |
+
|
| 55 |
#### Differences from v1.0.0 (10 months in the making)
|
| 56 |
- Noticeable improvements in quality and detail
|
| 57 |
- More variety in hairstyles, hair colors, eye colors, and even accessories (earrings and sunglasses)
|