Skip to content

feat(training): add standalone two-phase training - #18

Open
heiwang1997 wants to merge 1 commit into
NVIDIA:mainfrom
heiwang1997:feat/standalone-training
Open

feat(training): add standalone two-phase training#18
heiwang1997 wants to merge 1 commit into
NVIDIA:mainfrom
heiwang1997:feat/standalone-training

Conversation

@heiwang1997

Copy link
Copy Markdown
Contributor

Summary

  • add a standalone two-phase training pipeline for calibrated NCore V4 data, including example configurations, supervision, losses, differentiable rendering, optimizer scheduling, and documentation
  • add deterministic invalid-sample retry, distributed-safe experiment logging, and exact checkpoint/resume and preemption handling
  • add regression coverage for normalization, seeded sampling, renderer tensor contracts and post-processing order, and masked RGB loss

Validation

  • uv lock --check
  • ruff check --no-cache .
  • PYTHONDONTWRITEBYTECODE=1 .venv/bin/python -m pytest -p no:cacheprovider -q tests — 616 passed
  • completed one local single-GPU context step and one rendering step

Scope

This change supports the dense front-camera two-phase path. Other model
families and variable-camera curricula remain unsupported. The GPU smoke test
covers one step only; it does not establish renderer-kernel bitwise
equivalence, distributed convergence, or final model quality.

@heiwang1997 heiwang1997 mentioned this pull request Aug 24, 2026
@heiwang1997
heiwang1997 force-pushed the feat/standalone-training branch from ccf9eac to 924e968 Compare August 24, 2026 23:39
Signed-off-by: Jiahui Huang <jiahuih@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant