Rename 2023-03-29T12-33-22-project.yaml to config.yaml

2023-03-29T12-33-22-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: 480
@@ -23,17 +23,4 @@ model:
  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