Update config.yaml
Browse files- config.yaml +2 -3
config.yaml
CHANGED
|
@@ -2,7 +2,7 @@ model:
|
|
| 2 |
learning_rate: 0.0003
|
| 3 |
target: celle_main.CELLE_trainer
|
| 4 |
params:
|
| 5 |
-
ckpt_path:
|
| 6 |
condition_model_path:
|
| 7 |
condition_config_path: nucleus_vqgan.yaml
|
| 8 |
vqgan_model_path:
|
|
@@ -22,5 +22,4 @@ model:
|
|
| 22 |
fixed_embedding: true
|
| 23 |
text_embedding: esm2
|
| 24 |
loss_img_weight: 1
|
| 25 |
-
loss_cond_weight: 1
|
| 26 |
-
|
|
|
|
| 2 |
learning_rate: 0.0003
|
| 3 |
target: celle_main.CELLE_trainer
|
| 4 |
params:
|
| 5 |
+
ckpt_path: model.ckpt
|
| 6 |
condition_model_path:
|
| 7 |
condition_config_path: nucleus_vqgan.yaml
|
| 8 |
vqgan_model_path:
|
|
|
|
| 22 |
fixed_embedding: true
|
| 23 |
text_embedding: esm2
|
| 24 |
loss_img_weight: 1
|
| 25 |
+
loss_cond_weight: 1
|
|
|