KavehMoradkhani's picture
Add verified SimCortex v2 pretrained model archive
63929b9 verified
Raw History Blame Contribute Delete
2.2 kB
user_config: null
dataset:
split_file: /path/to/datasets/splits/dataset_split.csv
train_split_name: train
val_split_name: val
session_label: '01'
space: MNI152
roots:
HCP_YA: /path/to/datasets/hcpya-u100/derivatives/sc-preproc
OASIS1: /path/to/datasets/oasis-1/derivatives/sc-preproc
initsurf_roots:
HCP_YA: /path/to/datasets/hcpya-u100/derivatives/sc-initsurf
OASIS1: /path/to/datasets/oasis-1/derivatives/sc-initsurf
surface_name:
- lh_pial
- lh_white
- rh_pial
- rh_white
prob_clip_min: 0.0
prob_clip_max: 1.0
prob_gamma: 1.0
aug_prob: 0.3
aug_rot_range_deg: 3.0
aug_scale_range: 0.02
aug_trans_range_mm: 1.0
aug_intensity_prob: 0.3
aug_bias_strength: 0.2
aug_gain_range: 0.1
aug_bright_range: 0.1
aug_noise_std: 0.05
trainer:
seed: 2025
img_batch_size: 1
grad_accum_steps: 4
num_epochs: 600
validation_interval: 5
collision_interval: 5
learning_rate: 1.0e-4
weight_decay: 1.0e-4
points_per_image: 180000
val_points_per_image: 150000
num_workers: 4
log_level: INFO
mesh_chunk: 4
grad_clip_norm: 1.0
scheduler_patience: 4
scheduler_factor: 0.5
scheduler_min_lr: 1.0e-6
scheduler_threshold_mm: 0.0004
scheduler_threshold_mode: abs
scheduler_cooldown: 1
early_stop_patience: 15
early_stop_min_delta_mm: 0.0003
resume_from: null
model:
sigma: 1
n_steps: 8
inshape:
- 184
- 224
- 184
c_hid:
- 8
- 16
- 32
- 64
- 128
- 128
c_in: 2
geom_ratio: 0.5
geom_depth: 6
gn_groups: 8
gate_init: -3.0
dropout: 0.1
init_ckpt: null
init_strict: true
objective:
chamfer_weight: 1.0
chamfer_scale: 1.0
edge_loss_weight: 2.5
normal_weight: 0.25
reg_warmup_epochs: 25
hd_weight: 5.5
hd_p: 0.05
hd_lambda_mm: 0.75
hd_points: 50000
signed_nested_weight: 8.0
signed_margin_mm: 0.55
signed_points: 50000
pial_lr_hd_weight: 1.5
pial_lr_hd_p: 0.05
pial_lr_hd_lambda_mm: 1.0
pial_lr_hd_points: 30000
checkpoint:
alpha_wp: 0.03
alpha_lr: 0.015
rmse_guardrail_rel: 1.03
min_delta_score: 0.0001
require_collision_for_best: true
outputs:
root: /path/to/simcortex-runs/deform/exp01_hcpya+oasis1