Rename 2023-03-02T14-28-53-project.yaml to config.yaml

2023-03-02T14-28-53-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-11-28T21-06-56_OpenCell_nucleus_vqgan/checkpoints/epoch=000482.ckpt
-    condition_config_path: logs/2022-11-28T21-06-56_OpenCell_nucleus_vqgan/configs/2022-11-28T21-06-56-project.yaml
-    vqgan_model_path: logs/2023-01-31T18-43-53_OpenCell_threshold_vqgan/checkpoints/last.ckpt
-    vqgan_config_path: logs/2023-01-31T18-43-53_OpenCell_threshold_vqgan/configs/2023-02-13T11-27-14-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: 640
@@ -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/OpenCell/train_test_split_HPA_50.csv
-    dataset: OpenCell
-    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