file upload

2023-03-13T09-48-01-project.yaml

@@ -0,0 +1,39 @@
+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

epoch=000001.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fdb08420b54fdcd1d0c05046d0446f756dbe7d020541feb11dcff0fd05a04b69
+size 17513584491