HuangLab
/

CELL-E_2_HPA_2560

Model card Files Files and versions Community

Emaad commited on May 15, 2023

Commit

b951c80

•

1 Parent(s): c1b8d18

Update config.yaml

Files changed (1) hide show

config.yaml +2 -2

config.yaml CHANGED Viewed

@@ -2,7 +2,7 @@ model:
  learning_rate: 0.0003
  target: celle_main.CELLE_trainer
  params:
- ckpt_path: null
  condition_model_path:
  condition_config_path: nucleus_vqgan.yaml
  vqgan_model_path:
@@ -22,4 +22,4 @@ model:
  fixed_embedding: true
  text_embedding: esm2
  loss_img_weight: 1
- loss_cond_weight: 1

  learning_rate: 0.0003
  target: celle_main.CELLE_trainer
  params:
+ ckpt_path: model.ckpt
  condition_model_path:
  condition_config_path: nucleus_vqgan.yaml
  vqgan_model_path:
  fixed_embedding: true
  text_embedding: esm2
  loss_img_weight: 1
+ loss_cond_weight: 1