2511
Browse files- samples/sample_decoded_0.jpg +3 -0
- samples/sample_decoded_1.jpg +3 -0
- samples/sample_decoded_2.jpg +3 -0
- samples/sample_real_0.jpg +3 -0
- samples/sample_real_1.jpg +3 -0
- samples/sample_real_2.jpg +3 -0
- train_vae_fdl.py +3 -3
- vae/diffusion_pytorch_model.safetensors +1 -1
samples/sample_decoded_0.jpg
ADDED
|
Git LFS Details
|
samples/sample_decoded_1.jpg
ADDED
|
Git LFS Details
|
samples/sample_decoded_2.jpg
ADDED
|
Git LFS Details
|
samples/sample_real_0.jpg
ADDED
|
Git LFS Details
|
samples/sample_real_1.jpg
ADDED
|
Git LFS Details
|
samples/sample_real_2.jpg
ADDED
|
Git LFS Details
|
train_vae_fdl.py
CHANGED
|
@@ -45,7 +45,7 @@ model_resolution = 256
|
|
| 45 |
high_resolution = 512
|
| 46 |
limit = 0
|
| 47 |
save_barrier = 1.3
|
| 48 |
-
warmup_percent = 0.
|
| 49 |
percentile_clipping = 99
|
| 50 |
beta2 = 0.997
|
| 51 |
eps = 1e-8
|
|
@@ -65,8 +65,8 @@ kl_ratio = 0.00
|
|
| 65 |
|
| 66 |
# Доли лоссов
|
| 67 |
loss_ratios = {
|
| 68 |
-
"lpips": 0.
|
| 69 |
-
"fdl" :
|
| 70 |
"edge": 0.05,
|
| 71 |
"mse": 0.10,
|
| 72 |
"mae": 0.10,
|
|
|
|
| 45 |
high_resolution = 512
|
| 46 |
limit = 0
|
| 47 |
save_barrier = 1.3
|
| 48 |
+
warmup_percent = 0.005
|
| 49 |
percentile_clipping = 99
|
| 50 |
beta2 = 0.997
|
| 51 |
eps = 1e-8
|
|
|
|
| 65 |
|
| 66 |
# Доли лоссов
|
| 67 |
loss_ratios = {
|
| 68 |
+
"lpips": 0.50,
|
| 69 |
+
"fdl" : 0.25,
|
| 70 |
"edge": 0.05,
|
| 71 |
"mse": 0.10,
|
| 72 |
"mae": 0.10,
|
vae/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 382598708
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5572d6603f973b4726278390b9a40d1478b30ee8f298574acfd2e2c663227e09
|
| 3 |
size 382598708
|