file upload

2023-03-29T12-33-22-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-11-28T21-06-56_OpenCell_nucleus_vqgan/checkpoints/epoch=000482.ckpt
+    condition_config_path: logs/2022-11-28T21-06-56_OpenCell_nucleus_vqgan/configs/2022-11-28T21-06-56-project.yaml
+    vqgan_model_path: logs/2023-01-31T18-43-53_OpenCell_threshold_vqgan/checkpoints/last.ckpt
+    vqgan_config_path: logs/2023-01-31T18-43-53_OpenCell_threshold_vqgan/configs/2023-02-13T11-27-14-project.yaml
+    image_key: threshold
+    num_images: 2
+    dim: 480
+    num_text_tokens: 33
+    text_seq_len: 1000
+    depth: 68
+    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

epoch=000035.ckpt

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