Rename 2023-04-07T11-47-13-project.yaml to config.yaml

2023-04-07T11-47-13-project.yaml

@@ -1,39 +0,0 @@
-model:
- learning_rate: 0.0003
- target: celle_main.CELLE_trainer
- params:
- ckpt_path: logs/2023-03-02T14-28-43_HPA_celle_ESM2_640/checkpoints/epoch=000013.ckpt
- condition_model_path: logs/2022-11-07T17-17-51_OpenCell_nucleus_vqgan_finetune/checkpoints/epoch=000468.ckpt
- condition_config_path: logs/2022-11-07T17-17-51_OpenCell_nucleus_vqgan_finetune/configs/2023-02-23T19-14-01-project.yaml
- vqgan_model_path: logs/2023-02-13T11-37-10_OpenCell_threshold_vqgan_finetune/checkpoints/last.ckpt
- vqgan_config_path: logs/2023-02-13T11-37-10_OpenCell_threshold_vqgan_finetune/configs/2023-02-13T11-37-10-project.yaml
- image_key: threshold
- num_images: 2
- dim: 640
- num_text_tokens: 33
- text_seq_len: 1000
- depth: 55
- heads: 16
- dim_head: 64
- attn_dropout: 0.1
- ff_dropout: 0.1
- 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

config.yaml

@@ -0,0 +1,25 @@
+model:
+ learning_rate: 0.0003
+ target: celle_main.CELLE_trainer
+ params:
+ 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
+ num_text_tokens: 33
+ text_seq_len: 1000
+ depth: 55
+ heads: 16
+ dim_head: 64
+ attn_dropout: 0.1
+ ff_dropout: 0.1
+ attn_types: full
+ rotary_emb: true
+ fixed_embedding: true
+ text_embedding: esm2
+ loss_img_weight: 1
+ loss_cond_weight: 1