Add stage 1 and stage 2 intermediate checkpoints
#1
by weikaih - opened
Release the two intermediate checkpoints from the 3-stage WildDet3D training pipeline (requested in issue #20):
- wilddet3d_stage1_omni3d_12e_v1.0.pt (~4.7 GB): Stage 1, Omni3D-only canonical rotation, 12 epochs, batch 128. Loadable with build_model(skip_pretrained=True) and feeds the stage 2 all-data finetune.
- wilddet3d_stage2_alldata_12e_v1.0.pt (~4.7 GB): Stage 2, 8-dataset (Omni3D + CA1M + Waymo + 3EED + FoundationPose + ITW human + V3Det human) canonical finetune from stage 1, 12 epochs. Feeds the stage 3 negative-loss finetune that produced the released v1.0 paper checkpoint.
Same format as the existing v1.0.pt release: state_dict + epoch/step + hparams; optimizer/scheduler state stripped.
weikaih changed pull request status to merged