Emaad commited on
Commit
c1b8d18
1 Parent(s): 0a9413b

Rename 2023-03-13T09-48-01-project.yaml to config.yaml

Browse files
2023-03-13T09-48-01-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: 2560
@@ -20,20 +20,6 @@ model:
20
  attn_types: full
21
  rotary_emb: true
22
  fixed_embedding: true
23
- monitor: val/loss_epoch
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: theshold_vqgan.yaml
10
  image_key: threshold
11
  num_images: 2
12
  dim: 2560
 
20
  attn_types: full
21
  rotary_emb: true
22
  fixed_embedding: true
 
23
  text_embedding: esm2
24
  loss_img_weight: 1
25
  loss_cond_weight: 1