NSP / config.json
Potestates's picture
Upload V3 strong-fusion Phase-1 checkpoint (500 LUNA16 cases)
d64e483 verified
Raw
History Blame Contribute Delete
1.77 kB
{
"phase1_root": "/anvil/scratch/x-tding2/data/nsp_r2gs_v3_250/phase1_data/luna500_phase1",
"synth_root": "/anvil/scratch/x-tding2/data/nsp_r2gs_v3_250/synthetic_dataset/luna500_cone_ntrain_25_angle_360",
"cache_root": "/anvil/scratch/x-tding2/data/nsp_r2gs_v3_250/phase1_data/_phase1_v3_cache_500",
"output_dir": "/anvil/scratch/x-tding2/nsp/NSP_R2GS/outputs/v3_strongfusion_warm500_20260716",
"max_cases": 500,
"seed": 42,
"val_ratio": 0.1,
"strict_pack": false,
"prepare_packs": true,
"epochs": 30,
"batch_size": 1,
"num_workers": 0,
"lr": 0.0001,
"weight_decay": 0.0001,
"eval_every": 2,
"save_every": 5,
"vol_size": 128,
"proj_size": 64,
"gap_proj_size": 128,
"num_sparse_views": 25,
"num_gap_views": 12,
"base_ch": 48,
"ch_mults": "1,2,4,4",
"num_res_blocks": 2,
"embed_dim": 256,
"view_feat_ch": 24,
"lift_feat_ch": 12,
"fusion_grid_size": 32,
"null_ctx_ch": 24,
"null_hidden_ch": 32,
"null_gain": 0.1,
"xray_base_enabled": 1,
"xray_base_feat_ch": 24,
"xray_token_dim": 48,
"xray_token_grid_size": 4,
"xray_num_heads": 4,
"xray_num_layers": 2,
"xray_guidance_hidden_ch": 16,
"xray_fusion_mode": "strong",
"xray_fusion_feat_ch": 32,
"use_null_branch": 0,
"base_ctx_ch": 24,
"grad_loss_weight": 0.2,
"base_l1_weight": 0.35,
"base_grad_weight": 0.05,
"null_fourier_weight": 0.08,
"null_fourier_warmup": 12,
"null_fourier_mode": "mse",
"null_fourier_freq_min": 0.0,
"null_fourier_freq_max": 0.35,
"obs_consistency_weight": 0.02,
"null_l1_weight": 0.02,
"pred_activation": "none",
"softclamp_tau": 0.05,
"resume": "/anvil/scratch/x-tding2/nsp/NSP_R2GS_pre_sync_20260714/diffusion_prior/checkpoints/v27_decouplednull_250/best.pth",
"resume_optimizer": false
}