{
  "data": {
    "seed": 42,
    "profile": "t4x2",
    "train_max_len": 256,
    "eval_max_len": 850,
    "min_len": 24,
    "train_targets": 1000000000,
    "calibration_fraction": 0.15,
    "cluster_jaccard": 0.62,
    "cluster_len_ratio": 0.72,
    "msa_max_sequences": 96,
    "template_bank_size": 420,
    "kmer": 3,
    "d_node": 80,
    "d_pair": 48,
    "n_node_layers": 3,
    "n_pair_blocks": 4,
    "dropout": 0.1,
    "batch_size": 2,
    "val_batch_size": 1,
    "ablation_epochs": 4,
    "geometry_epochs": 10,
    "geometry_lr": 0.0002,
    "weight_decay": 0.01,
    "grad_clip": 1.0,
    "patience": 3,
    "num_workers": 2,
    "use_amp": true,
    "epoch_val_structures": 32,
    "contact_cutoff": 12.0,
    "bp_distance_cutoff": 15.0,
    "stacking_cutoff": 8.0,
    "w_dist": 1.0,
    "w_contact": 0.15,
    "w_bp": 0.2,
    "w_orient": 0.1,
    "w_stack": 0.08,
    "w_router_entropy": 0.002,
    "local_pair_boost": 1.25,
    "p_drop_msa": 0.15,
    "p_drop_template": 0.25,
    "p_drop_bp": 0.15,
    "p_sequence_only": 0.08,
    "flow_d_hidden": 80,
    "flow_layers": 4,
    "flow_epochs": 6,
    "flow_batch_size": 4,
    "flow_lr": 0.0001,
    "flow_patience": 2,
    "flow_distance_weight": 0.1,
    "flow_template_drop": 0.5,
    "flow_noise_scale": 16.0,
    "flow_steps": 14,
    "calib_candidates_aware": 5,
    "calib_candidates_denovo": 5,
    "final_candidates_aware": 8,
    "final_candidates_denovo": 8,
    "candidate_micro_batch": 4,
    "refine_steps": 8,
    "refine_lr": 0.03,
    "selector_epochs": 120,
    "selector_lr": 0.002,
    "selector_mc_dropout": 8,
    "uncertainty_lambda": 0.1,
    "diversity_lambda": 0.15,
    "n_top": 5,
    "hard_infer_len": 850,
    "out_dir": "/kaggle/working/EVIRNA_NOVA_CONTINUE_V12"
  },
  "nova": {
    "profile": "t4",
    "seed": 42,
    "d_node": 160,
    "d_pair": 96,
    "d_msa": 64,
    "n_node_heads": 8,
    "n_pair_blocks": 8,
    "pair_transition_mult": 4,
    "triangle_rank": 24,
    "dropout": 0.1,
    "use_axial_attention": true,
    "axial_heads": 4,
    "axial_every": 3,
    "triangle_chunk": 32,
    "recycle_iters": 2,
    "gradient_checkpointing": true,
    "n_sources": 5,
    "router_hidden": 192,
    "router_temperature": 0.85,
    "evidential_router": true,
    "router_dropout": 0.1,
    "relpos_buckets": 64,
    "relpos_max": 1024,
    "n_dist_bins": 64,
    "dist_min": 2.0,
    "dist_max": 42.0,
    "contact_cutoff": 12.0,
    "stack_cutoff": 8.0,
    "flow_node_dim": 160,
    "flow_pair_dim": 64,
    "flow_blocks": 6,
    "flow_rbf_bins": 32,
    "flow_time_dim": 64,
    "flow_noise_scale": 18.0,
    "flow_train_template_prob": 0.5,
    "flow_template_noise": 3.0,
    "flow_steps_train": 1,
    "flow_steps_infer": 48,
    "flow_heun": true,
    "dynamic_flow_trust": true,
    "flow_coord_clip": 5.0,
    "n_conformers": 5,
    "p_drop_foundation": 0.08,
    "p_drop_msa": 0.15,
    "p_drop_template": 0.25,
    "p_drop_ss": 0.12,
    "w_distogram": 0.4,
    "w_distance": 0.3,
    "w_contact": 0.12,
    "w_bp": 0.2,
    "w_stack": 0.08,
    "w_orientation": 0.06,
    "w_pair_conf": 0.05,
    "w_flow": 1.0,
    "w_flow_distance": 0.15,
    "w_router_entropy": 0.002,
    "w_router_smooth": 0.002,
    "w_router_balance": 0.001,
    "w_uncertainty_calibration": 0.03,
    "w_teacher_distill": 0.1,
    "lr": 0.00015,
    "weight_decay": 0.05,
    "warmup_steps": 2000,
    "grad_clip": 1.0
  },
  "args": {
    "profile": "t4",
    "data_profile": "t4x2",
    "epochs": 0,
    "batch_size": 2,
    "grad_accum": 8,
    "val_batch_size": 1,
    "val_batches": 24,
    "train_targets": 0,
    "train_max_len": 256,
    "eval_max_len": 850,
    "crop_len": 256,
    "crop_stride": 192,
    "max_windows_per_target": 2,
    "rnet_extract_len": 192,
    "conformers": 5,
    "flow_steps": 48,
    "diversity_lambda": 0.1,
    "seed": 42,
    "workers": 2,
    "rnet_cache": null,
    "rnet_checkpoint": null,
    "rnet_single_dim": 128,
    "rnet_pair_dim": 16,
    "no_rnet_auto_download": false,
    "allow_internal_rna_proxy": false,
    "bp_cache": null,
    "resume": true,
    "resume_from": "/kaggle/input/datasets/leminhhung0101/arn3d-agent/EVIRNA_NOVA_CONTINUE",
    "resume_history": "/kaggle/input/datasets/leminhhung0101/arn3d-agent/EVIRNA_NOVA_CONTINUE/NOVA_history.csv",
    "skip_test": false,
    "single_gpu": false,
    "disable_gradient_checkpointing": false,
    "self_test": false,
    "eval_only": false,
    "out_dir": "/kaggle/working/EVIRNA_NOVA_CONTINUE_V12"
  }
}