Rename 2023-03-29T12-41-23-project.yaml to config.yaml

2023-03-29T12-41-23-project.yaml → config.yaml

@@ -2,11 +2,11 @@ model:
   learning_rate: 0.0003
   target: celle_main.CELLE_trainer
   params:
-    ckpt_path: null
-    condition_model_path: logs/2022-09-23T23-59-09_HPA_nucleus_vqgan/checkpoints/epoch=000108.ckpt
-    condition_config_path: logs/2022-09-23T23-59-09_HPA_nucleus_vqgan/configs/2022-09-23T23-59-09-project.yaml
-    vqgan_model_path: logs/2022-09-28T15-37-57_HPA_threshold_vqgan/checkpoints/last.ckpt
-    vqgan_config_path: logs/2022-09-28T15-37-57_HPA_threshold_vqgan/configs/2022-11-07T17-12-07-project.yaml
+    ckpt_path: model.ckpt
+    condition_model_path: 
+    condition_config_path: nucleus_vqgan.yaml
+    vqgan_model_path: 
+    vqgan_config_path: threshold_vqgan.yaml
     image_key: threshold
     num_images: 2
     dim: 1280
@@ -20,20 +20,6 @@ model:
     attn_types: full
     rotary_emb: true
     fixed_embedding: true
-    monitor: val/loss_epoch
     text_embedding: esm2
     loss_img_weight: 1
-    loss_cond_weight: 1
-data:
-  target: celle_main.CellDataModule
-  params:
-    data_csv: /home/emaad/CELL-E_2/data/HPA/train_test_split_50_pct_cluster.csv
-    dataset: HPA
-    sequence_mode: embedding
-    vocab: esm2
-    threshold: median
-    text_seq_len: 1000
-    crop_size: 256
-    resize: 600
-    batch_size: 1
-    num_workers: 8
+    loss_cond_weight: 1