Emaad commited on
Commit
d0481a1
1 Parent(s): d6ba283

Rename 2023-03-02T14-28-43-project.yaml to config.yaml

Browse files
2023-03-02T14-28-43-project.yaml → config.yaml RENAMED
@@ -3,10 +3,10 @@ model:
3
  target: celle_main.CELLE_trainer
4
  params:
5
  ckpt_path: null
6
- condition_model_path: logs/2022-09-23T23-59-09_HPA_nucleus_vqgan/checkpoints/epoch=000108.ckpt
7
- condition_config_path: logs/2022-09-23T23-59-09_HPA_nucleus_vqgan/configs/2022-09-23T23-59-09-project.yaml
8
- vqgan_model_path: logs/2022-09-28T15-37-57_HPA_threshold_vqgan/checkpoints/last.ckpt
9
- vqgan_config_path: logs/2022-09-28T15-37-57_HPA_threshold_vqgan/configs/2022-11-07T17-12-07-project.yaml
10
  image_key: threshold
11
  num_images: 2
12
  dim: 640
@@ -24,16 +24,3 @@ model:
24
  text_embedding: esm2
25
  loss_img_weight: 1
26
  loss_cond_weight: 1
27
- data:
28
- target: celle_main.CellDataModule
29
- params:
30
- data_csv: /home/emaad/CELL-E_2/data/HPA/train_test_split_50_pct_cluster.csv
31
- dataset: HPA
32
- sequence_mode: embedding
33
- vocab: esm2
34
- threshold: median
35
- text_seq_len: 1000
36
- crop_size: 256
37
- resize: 600
38
- batch_size: 1
39
- num_workers: 8
 
3
  target: celle_main.CELLE_trainer
4
  params:
5
  ckpt_path: null
6
+ condition_model_path:
7
+ condition_config_path: nucleus_vqgan.yaml
8
+ vqgan_model_path:
9
+ vqgan_config_path: threshold_vqgan.yaml
10
  image_key: threshold
11
  num_images: 2
12
  dim: 640
 
24
  text_embedding: esm2
25
  loss_img_weight: 1
26
  loss_cond_weight: 1