| 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 | |