|
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 |
|
image_key: threshold |
|
num_images: 2 |
|
dim: 2560 |
|
num_text_tokens: 33 |
|
text_seq_len: 1000 |
|
depth: 5 |
|
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/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 |
|
|