Spaces:
Sleeping
Sleeping
| 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 |