celldreamer/config/evaluate_config.yml

run_name: "Eval_CellDreamer_V1"
model_type: "celldreamer"
device: "mps"


data_path: "celldreamer/data/datasets"
checkpoint_path: "celldreamer/checkpoints/best.pth"
output_dir: "celldreamer/results"
output_filename: "test_metrics.json"

batch_size: 128
kl_scale: 0.01 # updated to match train_config to prevent posterior collapse


# MUST BE SAME AS TRAINIG CONFIG
num_genes: 2446
latent_dim: 50
rnn_dim: 32
learning_rate: 25e-6

enc_hidden_dims:
  - 256
  - 128

dec_hidden_dims:
  - 128
  - 256

weight_decay: 1e-3