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)
|