Initial NeurIPS 2026 release: backbones, flow heads, SciPlex3 Path-B teachers/students, paper-table result JSONs
6d96029 verified | { | |
| "runs": [ | |
| { | |
| "paper_row": "Norman seed 1", | |
| "dataset": "CRISPRa Norman", | |
| "run_dir": "runs/b200_norman_flow_e115_krrinit_s02_mask_30k", | |
| "backbone": "hf-assets/checkpoints/crispra_norman_r16/scFATE_epoch100_periodic.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/CRISPRa-norman/norman2019_gears_split.h5ad", | |
| "training_script": "scripts/train_flow.py", | |
| "seed": 1, | |
| "result_json": "experiments/results/fair_comparison/norman_rotation_vs_direct__flow__b200_norman_flow_e115_krrinit_s02_mask_30k_K128.json", | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_norman_flow_e115_krrinit_s02_mask_30k/flow_best.pt", | |
| "size_mb": 46.5, | |
| "files_in_dir": [ | |
| "config.json", | |
| "flow_best.pt", | |
| "flow_metrics.jsonl", | |
| "krr_prior.pkl", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_v_net_state_dict_present": true, | |
| "theta_dim": 192, | |
| "d_embed": 1824, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "sigma": 0.02, | |
| "n_steps": 20, | |
| "epoch": 28500, | |
| "loss": 0.000217468106711749, | |
| "pert_names": "<list len=108 sample=['1', '2', 'AHR', 'ARID1A', 'ARRDC3']>", | |
| "mv_gene_names": "<list len=104 sample=['AHR', 'ARID1A', 'ARRDC3', 'ATL1', 'BAK1']>", | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 1.0 | |
| }, | |
| "hparams_nested": {} | |
| }, | |
| { | |
| "paper_row": "Norman seed 2", | |
| "dataset": "CRISPRa Norman", | |
| "run_dir": "runs/b200_norman_flow_e115_krrinit_s02_mask_30k_seed2", | |
| "backbone": "hf-assets/checkpoints/crispra_norman_r16/scFATE_epoch100_periodic.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/CRISPRa-norman/norman2019_gears_split.h5ad", | |
| "training_script": "scripts/train_flow.py", | |
| "seed": 2, | |
| "result_json": "experiments/results/fair_comparison/norman_rotation_vs_direct__flow__b200_norman_flow_e115_krrinit_s02_mask_30k_seed2.json", | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_norman_flow_e115_krrinit_s02_mask_30k_seed2/flow_best.pt", | |
| "size_mb": 46.5, | |
| "files_in_dir": [ | |
| "config.json", | |
| "flow_best.pt", | |
| "flow_metrics.jsonl", | |
| "krr_prior.pkl", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_v_net_state_dict_present": true, | |
| "theta_dim": 192, | |
| "d_embed": 1824, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "sigma": 0.02, | |
| "n_steps": 20, | |
| "epoch": 29000, | |
| "loss": 0.0001306741796724964, | |
| "pert_names": "<list len=108 sample=['1', '2', 'AHR', 'ARID1A', 'ARRDC3']>", | |
| "mv_gene_names": "<list len=104 sample=['AHR', 'ARID1A', 'ARRDC3', 'ATL1', 'BAK1']>", | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 1.0 | |
| }, | |
| "hparams_nested": {} | |
| }, | |
| { | |
| "paper_row": "Norman seed 3", | |
| "dataset": "CRISPRa Norman", | |
| "run_dir": "runs/b200_norman_flow_e115_krrinit_s02_mask_30k_seed3", | |
| "backbone": "hf-assets/checkpoints/crispra_norman_r16/scFATE_epoch100_periodic.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/CRISPRa-norman/norman2019_gears_split.h5ad", | |
| "training_script": "scripts/train_flow.py", | |
| "seed": 3, | |
| "result_json": "experiments/results/fair_comparison/norman_rotation_vs_direct__flow__b200_norman_flow_e115_krrinit_s02_mask_30k_seed3.json", | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_norman_flow_e115_krrinit_s02_mask_30k_seed3/flow_best.pt", | |
| "size_mb": 46.5, | |
| "files_in_dir": [ | |
| "config.json", | |
| "flow_best.pt", | |
| "flow_metrics.jsonl", | |
| "krr_prior.pkl", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_v_net_state_dict_present": true, | |
| "theta_dim": 192, | |
| "d_embed": 1824, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "sigma": 0.02, | |
| "n_steps": 20, | |
| "epoch": 29500, | |
| "loss": 0.0001756271449266933, | |
| "pert_names": "<list len=108 sample=['1', '2', 'AHR', 'ARID1A', 'ARRDC3']>", | |
| "mv_gene_names": "<list len=104 sample=['AHR', 'ARID1A', 'ARRDC3', 'ATL1', 'BAK1']>", | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 1.0 | |
| }, | |
| "hparams_nested": {} | |
| }, | |
| { | |
| "paper_row": "RPE1 seed 1", | |
| "dataset": "Replogle RPE1", | |
| "run_dir": "runs/b200_rpe1_flow_block_krrinit_mask_30k_s1", | |
| "backbone": "hf-assets/checkpoints/rpe1/scFATE_epoch350_periodic.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/replogle_rpe1/replogle_rpe1.h5ad", | |
| "training_script": "scripts/train_flow.py", | |
| "seed": 1, | |
| "result_json": "experiments/results/fair_comparison/rpe1_rotation_vs_direct__flow__b200_rpe1_flow_block_krrinit_mask_30k_s1_rpe1_block_K128.json", | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_rpe1_flow_block_krrinit_mask_30k_s1/flow_best.pt", | |
| "size_mb": 46.6, | |
| "files_in_dir": [ | |
| "config.json", | |
| "flow_best.pt", | |
| "flow_metrics.jsonl", | |
| "krr_prior.pkl", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_v_net_state_dict_present": true, | |
| "theta_dim": 192, | |
| "d_embed": 1824, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "sigma": 0.05, | |
| "n_steps": 20, | |
| "epoch": 28500, | |
| "loss": 5.099160883522321e-05, | |
| "pert_names": "<list len=1544 sample=['AARS', 'AASDHPPT', 'AATF', 'ABCB10', 'ABCB7']>", | |
| "mv_gene_names": "<list len=1524 sample=['AARS', 'AASDHPPT', 'AATF', 'ABCB10', 'ABCB7']>", | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 1.0 | |
| }, | |
| "hparams_nested": {} | |
| }, | |
| { | |
| "paper_row": "K562 base flow (teacher for reflow)", | |
| "dataset": "Replogle K562", | |
| "run_dir": "runs/b200_k562_flow_bs2048_krrinit_mask_30k", | |
| "backbone": "hf-assets/checkpoints/crispri_k562/scFATE_epoch700_periodic.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/replogle_k562/replogle_k562.h5ad", | |
| "training_script": "scripts/train_flow.py", | |
| "seed": 1, | |
| "result_json": "experiments/results/fair_comparison/replogle_rotation_vs_direct__flow__b200_k562_flow_bs2048_krrinit_mask_30k_K128.json", | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": "runs/launch_tx_bch_combo.sh", | |
| "checkpoint": { | |
| "path": "runs/b200_k562_flow_bs2048_krrinit_mask_30k/flow_best.pt", | |
| "size_mb": 46.5, | |
| "files_in_dir": [ | |
| "config.json", | |
| "flow_best.pt", | |
| "flow_metrics.jsonl", | |
| "krr_prior.pkl", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_v_net_state_dict_present": true, | |
| "theta_dim": 192, | |
| "d_embed": 1824, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "sigma": 0.05, | |
| "n_steps": 20, | |
| "epoch": 28000, | |
| "loss": 3.408003710949113e-05, | |
| "pert_names": "<list len=1093 sample=['AAMP', 'AARS', 'AATF', 'ABCB7', 'ABCE1']>", | |
| "mv_gene_names": "<list len=1085 sample=['AAMP', 'AARS', 'AATF', 'ABCB7', 'ABCE1']>", | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 1.0 | |
| }, | |
| "hparams_nested": {} | |
| }, | |
| { | |
| "paper_row": "K562 reflow K=2 (paper headline 81.2)", | |
| "dataset": "Replogle K562", | |
| "run_dir": "runs/b200_k562_flow_bs2048_krrinit_mask_30k_reflow_K2_nomask_bracket1ep0_s1", | |
| "backbone": "hf-assets/checkpoints/crispri_k562/scFATE_epoch700_periodic.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/replogle_k562/replogle_k562.h5ad", | |
| "training_script": "scripts/reflow_k562_delta_flow.py", | |
| "seed": 1, | |
| "result_json": "experiments/results/fair_comparison/replogle_rotation_vs_direct__flow__b200_k562_flow_bs2048_krrinit_mask_30k_reflow_K2_nomask_bracket1ep0_s1_reflow_K2_bracket1p0_ens5seed_sigmainf0p15_antithetic_Kper128.json", | |
| "teacher": "runs/b200_k562_flow_bs2048_krrinit_mask_30k", | |
| "teachers": null, | |
| "launch_script": "runs/launch_iter174_ens5_Kper256_and_train_s6_s10.sh", | |
| "checkpoint": { | |
| "path": "runs/b200_k562_flow_bs2048_krrinit_mask_30k_reflow_K2_nomask_bracket1ep0_s1/flow_best.pt", | |
| "size_mb": 46.5, | |
| "files_in_dir": [ | |
| "config.json", | |
| "flow_best.pt", | |
| "flow_metrics.jsonl", | |
| "krr_prior.pkl", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_v_net_state_dict_present": true, | |
| "theta_dim": 192, | |
| "d_embed": 1824, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "sigma": 0.05, | |
| "n_steps": 20, | |
| "epoch": 21000, | |
| "loss": 0.002288587740622461, | |
| "pert_names": "<list len=1093 sample=['AAMP', 'AARS', 'AATF', 'ABCB7', 'ABCE1']>", | |
| "mv_gene_names": "<list len=1085 sample=['AAMP', 'AARS', 'AATF', 'ABCB7', 'ABCE1']>", | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 1.0, | |
| "head_kind": "flat", | |
| "d_block_head": 16, | |
| "velocity_arch": "mlp", | |
| "teacher_arch": "mlp", | |
| "tx_d_token": 128, | |
| "tx_n_layers": 4, | |
| "tx_n_heads": 4, | |
| "tx_mlp_mult": 4.0, | |
| "cfg_drop": 0.0, | |
| "rectified_from": "runs/b200_k562_flow_bs2048_krrinit_mask_30k/flow_best.pt", | |
| "rectified_K": 2, | |
| "rectified_teacher_n_steps": 20, | |
| "target_noise_sigma": 0.0, | |
| "t_sampling": "uniform", | |
| "bch_aug_rate": 0.0, | |
| "bch_embed_mode": "mean", | |
| "bracket_reg": 1.0 | |
| }, | |
| "hparams_nested": {} | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priornone teacher s1", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 1, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 1, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "none", | |
| "krr_gamma": null, | |
| "krr_alpha": null, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priorkrr teacher s1", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 1, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 1, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 0.1, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priornone teacher s2", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 2, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 2, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "none", | |
| "krr_gamma": null, | |
| "krr_alpha": null, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priorkrr teacher s2", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 2, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 2, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 0.1, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priornone teacher s3", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 3, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 3, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "none", | |
| "krr_gamma": null, | |
| "krr_alpha": null, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priorkrr teacher s3", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 3, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 3, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 0.1, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priornone teacher s4", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 4, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 4, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "none", | |
| "krr_gamma": null, | |
| "krr_alpha": null, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priorkrr teacher s4", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 4, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 4, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 0.1, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priornone teacher s5", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 5, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 5, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "none", | |
| "krr_gamma": null, | |
| "krr_alpha": null, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priorkrr teacher s5", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 5, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 5, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 0.1, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priornone teacher s6", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 6, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": "runs/launch_iter186_sciplex3_delta_flow_reflow_K2_from_priornone_s6.sh", | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 6, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "none", | |
| "krr_gamma": null, | |
| "krr_alpha": null, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priorkrr teacher s6", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 6, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 6, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 0.1, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priornone teacher s7", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 7, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 7, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "none", | |
| "krr_gamma": null, | |
| "krr_alpha": null, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priorkrr teacher s7", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 7, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 7, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 0.1, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priornone teacher s8", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 8, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 8, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "none", | |
| "krr_gamma": null, | |
| "krr_alpha": null, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priorkrr teacher s8", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 8, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 8, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 0.1, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priornone teacher s9", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 9, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 9, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "none", | |
| "krr_gamma": null, | |
| "krr_alpha": null, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 priorkrr teacher s9", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/train_sciplex3_delta_flow.py", | |
| "seed": 9, | |
| "result_json": null, | |
| "teacher": null, | |
| "teachers": null, | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9", | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "seed": 9, | |
| "log_every": 1000, | |
| "topk_de": 50, | |
| "zscore_target": true, | |
| "prior": "krr", | |
| "krr_gamma": 0.01, | |
| "krr_alpha": 0.1, | |
| "dose_aware": false, | |
| "bch_aug": false, | |
| "bch_n_pairs": 0, | |
| "device": "cuda" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 mixed-18 K=16 student s1 (paper headline 70.0)", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s1", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/reflow_sciplex3_delta_flow_ensemble.py", | |
| "seed": 1, | |
| "result_json": "experiments/results/sciplex3_iter196_reflow_ensemble_mixed18_K16_N5.json", | |
| "teacher": null, | |
| "teachers": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9" | |
| ], | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s1/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "teacher_ckpts": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9/flow_best.pt" | |
| ], | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s1", | |
| "reflow_K": 16, | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "teacher_sigma": 0.3, | |
| "teacher_steps": 40, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "seed": 1, | |
| "log_every": 2000, | |
| "topk_de": 50, | |
| "device": "cuda", | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "M_teachers": 18, | |
| "teacher_prior_types": [ | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr" | |
| ], | |
| "prior": "none" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 mixed-18 K=16 student s2 (paper headline 70.0)", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s2", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/reflow_sciplex3_delta_flow_ensemble.py", | |
| "seed": 2, | |
| "result_json": "experiments/results/sciplex3_iter196_reflow_ensemble_mixed18_K16_N5.json", | |
| "teacher": null, | |
| "teachers": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9" | |
| ], | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s2/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "teacher_ckpts": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9/flow_best.pt" | |
| ], | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s2", | |
| "reflow_K": 16, | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "teacher_sigma": 0.3, | |
| "teacher_steps": 40, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "seed": 2, | |
| "log_every": 2000, | |
| "topk_de": 50, | |
| "device": "cuda", | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "M_teachers": 18, | |
| "teacher_prior_types": [ | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr" | |
| ], | |
| "prior": "none" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 mixed-18 K=16 student s3 (paper headline 70.0)", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s3", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/reflow_sciplex3_delta_flow_ensemble.py", | |
| "seed": 3, | |
| "result_json": "experiments/results/sciplex3_iter196_reflow_ensemble_mixed18_K16_N5.json", | |
| "teacher": null, | |
| "teachers": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9" | |
| ], | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s3/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "teacher_ckpts": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9/flow_best.pt" | |
| ], | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s3", | |
| "reflow_K": 16, | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "teacher_sigma": 0.3, | |
| "teacher_steps": 40, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "seed": 3, | |
| "log_every": 2000, | |
| "topk_de": 50, | |
| "device": "cuda", | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "M_teachers": 18, | |
| "teacher_prior_types": [ | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr" | |
| ], | |
| "prior": "none" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 mixed-18 K=16 student s4 (paper headline 70.0)", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s4", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/reflow_sciplex3_delta_flow_ensemble.py", | |
| "seed": 4, | |
| "result_json": "experiments/results/sciplex3_iter196_reflow_ensemble_mixed18_K16_N5.json", | |
| "teacher": null, | |
| "teachers": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9" | |
| ], | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s4/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "teacher_ckpts": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9/flow_best.pt" | |
| ], | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s4", | |
| "reflow_K": 16, | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "teacher_sigma": 0.3, | |
| "teacher_steps": 40, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "seed": 4, | |
| "log_every": 2000, | |
| "topk_de": 50, | |
| "device": "cuda", | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "M_teachers": 18, | |
| "teacher_prior_types": [ | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr" | |
| ], | |
| "prior": "none" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 mixed-18 K=16 student s5 (paper headline 70.0)", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s5", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/reflow_sciplex3_delta_flow_ensemble.py", | |
| "seed": 5, | |
| "result_json": "experiments/results/sciplex3_iter196_reflow_ensemble_mixed18_K16_N5.json", | |
| "teacher": null, | |
| "teachers": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9" | |
| ], | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s5/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "teacher_ckpts": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9/flow_best.pt" | |
| ], | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s5", | |
| "reflow_K": 16, | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "teacher_sigma": 0.3, | |
| "teacher_steps": 40, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "seed": 5, | |
| "log_every": 2000, | |
| "topk_de": 50, | |
| "device": "cuda", | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "M_teachers": 18, | |
| "teacher_prior_types": [ | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr" | |
| ], | |
| "prior": "none" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 mixed-18 K=16 student s6 (paper headline 70.0)", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s6", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/reflow_sciplex3_delta_flow_ensemble.py", | |
| "seed": 6, | |
| "result_json": "experiments/results/sciplex3_iter197_reflow_ensemble_mixed18_K16_N7.json", | |
| "teacher": null, | |
| "teachers": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9" | |
| ], | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s6/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "teacher_ckpts": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9/flow_best.pt" | |
| ], | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s6", | |
| "reflow_K": 16, | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "teacher_sigma": 0.3, | |
| "teacher_steps": 40, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "seed": 6, | |
| "log_every": 2000, | |
| "topk_de": 50, | |
| "device": "cuda", | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "M_teachers": 18, | |
| "teacher_prior_types": [ | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr" | |
| ], | |
| "prior": "none" | |
| } | |
| }, | |
| { | |
| "paper_row": "SciPlex3 mixed-18 K=16 student s7 (paper headline 70.0)", | |
| "dataset": "SciPlex3", | |
| "run_dir": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s7", | |
| "backbone": "hf-assets/checkpoints/sciplex3/scFATE_epoch199_best.pt", | |
| "h5ad_rel": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "training_script": "scripts/reflow_sciplex3_delta_flow_ensemble.py", | |
| "seed": 7, | |
| "result_json": "experiments/results/sciplex3_iter197_reflow_ensemble_mixed18_K16_N7.json", | |
| "teacher": null, | |
| "teachers": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9" | |
| ], | |
| "launch_script": null, | |
| "checkpoint": { | |
| "path": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s7/flow_best.pt", | |
| "size_mb": 54.7, | |
| "files_in_dir": [ | |
| "config.json", | |
| "eval.json", | |
| "flow_best.pt", | |
| "reproduce.sh" | |
| ] | |
| }, | |
| "hparams_top": { | |
| "_state_dict_present": true, | |
| "theta_dim": 2000, | |
| "d_embed": 2410, | |
| "y_mean": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "y_std": "<tensor shape=[1, 2000] dtype=torch.float32>", | |
| "train_names": "<list len=161 sample=['2-Methoxyestradiol', 'A-366', 'ABT-737', 'AG-490', 'AMG-900']>" | |
| }, | |
| "hparams_nested": { | |
| "teacher_ckpts": [ | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priornone_V2B_s9/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s1/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s2/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s3/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s4/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s5/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s6/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s7/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s8/flow_best.pt", | |
| "runs/b200_sciplex3_delta_flow_mv_v2_sig0p3_priorkrr_V2B_s9/flow_best.pt" | |
| ], | |
| "dataset_h5ad": "datasets/scFATE/processed/sciplex3/sciplex3.h5ad", | |
| "multiview": "data/drug_embeddings/sciplex3_multiview_v2.pt", | |
| "ood_json": "data/sciplex3_split_v2b.json", | |
| "output": "runs/b200_sciplex3_delta_flow_reflow_ensemble_mixed18_K16_V2B_s7", | |
| "reflow_K": 16, | |
| "epochs": 30000, | |
| "lr": 0.0003, | |
| "wd": 0.0001, | |
| "batch": 64, | |
| "sigma": 0.3, | |
| "teacher_sigma": 0.3, | |
| "teacher_steps": 40, | |
| "sigma_inf": 0.1, | |
| "n_steps": 20, | |
| "K_eval": 128, | |
| "seed": 7, | |
| "log_every": 2000, | |
| "topk_de": 50, | |
| "device": "cuda", | |
| "d_hidden": 512, | |
| "n_blocks": 4, | |
| "dropout": 0.1, | |
| "M_teachers": 18, | |
| "teacher_prior_types": [ | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "none", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr", | |
| "krr" | |
| ], | |
| "prior": "none" | |
| } | |
| } | |
| ] | |
| } |