Hugging Face's logo

SimCortexAI
/
simcortex-v2

PyTorch
English
medical-imaging
neuroimaging
brain-mri
cortical-surface-reconstruction
3d-mesh
Model card Files
xet
 Community
simcortex-v2
File size: 2,203 Bytes
63929b9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
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