alma / docs /finetuning.md
goel65's picture
Upload folder using huggingface_hub
aa51bb4 verified
|
Raw History Blame Contribute Delete
1.43 kB

Fine-tuning and evaluation

The final decoder uses metric local refinement (metric_local_refinement), with 128 local channels, two blocks, and at most 16 neighbors. Both schedules run up to 120 epochs, including ten frozen-encoder epochs, followed by cosine fine-tuning. The QL1 and QL2 trainers preserve their respective research protocols.

Copy configs/ql*_segmentation.yaml, edit data paths and choose a new output directory. Download the paired encoder artifact before fine-tuning. A complete segmentation checkpoint is sufficient for inference.

python scripts/finetune_segmentation.py --config configs/ql2_segmentation.yaml
python scripts/evaluate_segmentation.py --config configs/ql2_segmentation_eval.yaml

The evaluation configs load released best segmentation checkpoints and use FP32. When evaluating a new run, update model.segmentation_checkpoint. QL1 includes the final common-support evaluation seed. mIoU uses five unthresholded semantic classes with unmapped labels ignored. Confidence thresholding belongs to exported products. QL2 evaluation is on the thinned QL1 proxy. Crop visualization LAS files cannot reconstruct an original LAS header or CRS from NumPy input.

Record any change to support sampling, normalization, labels, physical radii, checkpoint, or split as a new experiment. Matching architecture alone does not reproduce scores: original splits and support selection also matter.