File size: 1,300 Bytes
8759e59 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
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
|