Update nucleus_vqgan.yaml
Browse files- nucleus_vqgan.yaml +1 -12
nucleus_vqgan.yaml
CHANGED
|
@@ -30,15 +30,4 @@ model:
|
|
| 30 |
disc_in_channels: 1
|
| 31 |
disc_start: 50000
|
| 32 |
disc_weight: 0.2
|
| 33 |
-
codebook_weight: 1.0
|
| 34 |
-
|
| 35 |
-
data:
|
| 36 |
-
target: celle_main.CellDataModule
|
| 37 |
-
params:
|
| 38 |
-
data_csv: /data/HPA/train_test_split_50_pct_cluster.csv
|
| 39 |
-
dataset: HPA
|
| 40 |
-
text_seq_len: 0
|
| 41 |
-
crop_size: 256
|
| 42 |
-
resize: 600
|
| 43 |
-
batch_size: 7
|
| 44 |
-
num_workers: 8
|
|
|
|
| 30 |
disc_in_channels: 1
|
| 31 |
disc_start: 50000
|
| 32 |
disc_weight: 0.2
|
| 33 |
+
codebook_weight: 1.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|