Update threshold_vqgan.yaml
Browse files- threshold_vqgan.yaml +1 -14
threshold_vqgan.yaml
CHANGED
|
@@ -3,7 +3,6 @@ model:
|
|
| 3 |
target: taming.models.vqgan.VQModel
|
| 4 |
params:
|
| 5 |
image_key: threshold
|
| 6 |
-
monitor: val/rec_loss
|
| 7 |
ckpt_path:
|
| 8 |
embed_dim: 256
|
| 9 |
n_embed: 512
|
|
@@ -31,16 +30,4 @@ model:
|
|
| 31 |
disc_in_channels: 1
|
| 32 |
disc_start: 100000
|
| 33 |
disc_weight: 0.2
|
| 34 |
-
codebook_weight: 1.0
|
| 35 |
-
|
| 36 |
-
data:
|
| 37 |
-
target: celle_main.CellDataModule
|
| 38 |
-
params:
|
| 39 |
-
data_csv: data/HPA/train_test_split_50_pct_cluster.csv
|
| 40 |
-
dataset: HPA
|
| 41 |
-
text_seq_len: 0
|
| 42 |
-
threshold: median
|
| 43 |
-
crop_size: 256
|
| 44 |
-
resize: 600
|
| 45 |
-
batch_size: 7
|
| 46 |
-
num_workers: 8
|
|
|
|
| 3 |
target: taming.models.vqgan.VQModel
|
| 4 |
params:
|
| 5 |
image_key: threshold
|
|
|
|
| 6 |
ckpt_path:
|
| 7 |
embed_dim: 256
|
| 8 |
n_embed: 512
|
|
|
|
| 30 |
disc_in_channels: 1
|
| 31 |
disc_start: 100000
|
| 32 |
disc_weight: 0.2
|
| 33 |
+
codebook_weight: 1.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|