Files

DRF Author Checkpoint Artifact

This artifact bundle archives the author-provided DRF checkpoint together with the compatibility configs that make it usable in this repo.

Files:

  • DRF_118_unordered_iter2w_lr0.001_8830-08000.pt
  • author_original_config.yaml
  • eval_112_1gpu.yaml
  • eval_118_defaultroot_1gpu.yaml
  • eval_118_splitroot_1gpu.yaml
  • eval_118_aligned_1gpu.yaml
  • eval_118_paper_1gpu.yaml

Best recovered path:

  • config: configs/drf/drf_author_eval_118_aligned_1gpu.yaml
  • result:
    • 80.24 Acc / 76.73 Prec / 76.40 Rec / 76.56 F1

Run Commands

Recommended eval:

CUDA_VISIBLE_DEVICES=GPU-9cc7b26e-90d4-0c49-4d4c-060e528ffba6 \
uv run torchrun --nproc_per_node=1 --master_port=29693 \
  opengait/main.py \
  --cfgs ./configs/drf/drf_author_eval_118_aligned_1gpu.yaml \
  --phase test

Other compatibility checks:

CUDA_VISIBLE_DEVICES=GPU-9cc7b26e-90d4-0c49-4d4c-060e528ffba6 \
uv run torchrun --nproc_per_node=1 --master_port=29695 \
  opengait/main.py \
  --cfgs ./configs/drf/drf_author_eval_112_1gpu.yaml \
  --phase test

CUDA_VISIBLE_DEVICES=GPU-9cc7b26e-90d4-0c49-4d4c-060e528ffba6 \
uv run torchrun --nproc_per_node=1 --master_port=29696 \
  opengait/main.py \
  --cfgs ./configs/drf/drf_author_eval_118_splitroot_1gpu.yaml \
  --phase test

CUDA_VISIBLE_DEVICES=GPU-9cc7b26e-90d4-0c49-4d4c-060e528ffba6 \
uv run torchrun --nproc_per_node=1 --master_port=29697 \
  opengait/main.py \
  --cfgs ./configs/drf/drf_author_eval_118_paper_1gpu.yaml \
  --phase test

Paths To Change On Another Machine

If you move this artifact bundle or use a different dataset location, update:

  • data_cfg.dataset_root
  • data_cfg.dataset_partition
  • evaluator_cfg.restore_hint

In this repo, the checkpoint is archived at:

  • artifact/scoliosis_drf_author_118_compat/DRF_118_unordered_iter2w_lr0.001_8830-08000.pt