1.8 KiB
1.8 KiB
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.ptauthor_original_config.yamleval_112_1gpu.yamleval_118_defaultroot_1gpu.yamleval_118_splitroot_1gpu.yamleval_118_aligned_1gpu.yamleval_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_rootdata_cfg.dataset_partitionevaluator_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