[TAO-2511][Feature] Support frozen DINOv3 backbones in Visual ChangeNet - #95
Conversation
|
For security reasons, CI does not run automatically on NVIDIA's runners — it must be triggered per commit.
Tip Should this fix also ship in a release? Add a Important Before merging: every file needs a license header and every commit must be DCO signed-off — see the 📖 Contribution guide for details. |
|
/build |
vpraveen-nv
left a comment
There was a problem hiding this comment.
LGTM — the 7B wiring (40-block partition, 4096 channel maps) checks out against timm, the raw-SSL rejection matches the converter's remap table, and the C-RADIO/FAN paths are untouched. A few non-blocking questions inline. One sequencing note: please land the tao-core companion (NVIDIA-TAO/tao-core#31) in lockstep so schema validation doesn't reject vit_7b_dinov3 in the window between merges.
|
/build |
Signed-off-by: Nikhil <nnagrajrao@nvidia.com>
5f03cfb to
8cbbcf6
Compare
|
/build |
1 similar comment
|
/build |
What changes are proposed in this pull request?
Add frozen DINOv3 backbone support to the existing Visual ChangeNet classification and segmentation paths with minimal changes to the AOI architecture.
Example:
Set
pretrained_backbone_path: nullto use the matching timm/Hugging Face pretrained weights.Why are the changes needed?
DEFT AOI needs to reuse DINOv3 SSL features without updating the backbone. This keeps the established Visual ChangeNet task-level fine-tuning design while allowing either a converted TAO DINOv3 checkpoint or the compatible public timm/Hugging Face weights.
Related issues
JIRA: TAO-2511.
Companion changes are in tao-core and tao-skill-bank on the same TAO-2511 branch.
Does this PR introduce any user-facing change?
Yes. Visual ChangeNet users can select all six DINOv3 variants and freeze the ViT. Existing backbone types, C-RADIO behavior, and defaults are unchanged.
How was this patch tested?
git diff --checkand no-Markdown checks passed.Was this patch authored or co-authored using generative AI tooling?
Generated-by: OpenAI Codex
Release note
Checklist
git commit -s)Notes for reviewers
No Markdown files are changed. Start with
cv/backbone_v2/dino_v3.py, then the two Visual ChangeNet model constructors and the focused regression test. The change deliberately reuses existing AOI adapters and heads instead of introducing a parallel task package.