Update nucleus_vqgan.yaml
Browse files- nucleus_vqgan.yaml +1 -10
nucleus_vqgan.yaml
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
model:
|
| 2 |
-
base_learning_rate: 4.5e-06
|
| 3 |
target: taming.models.vqgan.VQModel
|
| 4 |
params:
|
| 5 |
image_key: nucleus
|
|
@@ -22,12 +21,4 @@ model:
|
|
| 22 |
num_res_blocks: 2
|
| 23 |
attn_resolutions:
|
| 24 |
- 16
|
| 25 |
-
dropout: 0.0
|
| 26 |
-
lossconfig:
|
| 27 |
-
target: taming.modules.losses.vqperceptual.VQLPIPSWithDiscriminator
|
| 28 |
-
params:
|
| 29 |
-
disc_conditional: false
|
| 30 |
-
disc_in_channels: 1
|
| 31 |
-
disc_start: 50000
|
| 32 |
-
disc_weight: 0.2
|
| 33 |
-
codebook_weight: 1.0
|
|
|
|
| 1 |
model:
|
|
|
|
| 2 |
target: taming.models.vqgan.VQModel
|
| 3 |
params:
|
| 4 |
image_key: nucleus
|
|
|
|
| 21 |
num_res_blocks: 2
|
| 22 |
attn_resolutions:
|
| 23 |
- 16
|
| 24 |
+
dropout: 0.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|