Nodes/ComfyUI-NeurCADRecon/NeurCADRecon Train
ComfyUI Node

NeurCADRecon Train

Train NeurCADRecon to learn an SDF from a point cloud. Loss Terms (shown during training): • SDF: Distance error at surface points. Good: < 0.01 • Eikonal: Gradient magnitude should be 1 (|∇f|=1). Good: < 0.1 • Morse: Gaussian curvature regularization for sharp CAD edges. Good: < 1.0 • Total: Weighted sum. Typically starts 100-500, ends 10-50. Training takes 5-15 min on GPU for 10k iterations.

By PozzettiAndrea-archive·Created 8 months ago·Updated 6 months ago· 0
NeurCADRecon Train
  • model
  • point_cloud
  • trained_model
  • checkpoint_path
  • training_log
num_iterations10000
batch_size20000
learning_rate0.0001
loss_presetbalanced
save_checkpointtrue
checkpoint_dir
log_interval500
CategoryNeurCADRecon

Inputs (9)

NameTypeDefaultDescription
modelNEURCADRECON_MODEL
point_cloudTRIMESH
num_iterationsoptINT100001000–50000Number of training iterations. 10000 is typical for good quality.
batch_sizeoptINT200005000–50000Number of points sampled per iteration. Higher uses more GPU memory.
learning_rateoptFLOAT0.00010.000001–0.001Learning rate for Adam optimizer.
loss_presetoptCOMBObalancedLoss weight preset. 'sharp_edges' increases morse weight for CAD-like shapes.
save_checkpointoptBOOLEANtrueSave checkpoint after training for later use.
checkpoint_diroptSTRINGDirectory to save checkpoint. Empty uses default location.
log_intervaloptINT500100–2000Print loss every N iterations.

Outputs (3)

NameTypeDescription
trained_modelNEURCADRECON_MODEL
checkpoint_pathSTRING
training_logSTRING