Skip to content

Migrate timm_mobilevit to the semantic Task SDK #1526

Closed

Description

@yifeif-nv

Parent epic: #1430

Status: Not started — comment here before beginning work.

Family: families/timm_mobilevit/
Current declared task mode(s): classification
Target semantic Task(s): image_to_class_scores

Related family CLI work: #1375. It is a separate change and does not complete this Task migration.

Work

  • Inventory this family's primary and secondary behavior, including batch, streaming, raw outputs and stateful operations where supported. Record the current defaults, profiles, checkpoint cases, and reference criteria.
  • Implement and bind the matching semantic Task interface(s) in this family's runtime. Preserve its graph, preprocessing, inference, postprocessing, configuration and complete owned results.
  • Align family-owned support, build metadata and manifests with the primary Task. Advertise only interfaces implemented by the loaded bundle; document whether existing bundles need rebuilding.
  • Add family-owned CPU contracts and public C/C++ SDK consumers. Cover invalid inputs, configuration handling, result lifetime and every relevant lifecycle.
  • Run a selected real checkpoint E2E with the existing oracle and thresholds, then check the remaining declared capabilities and public CLI, example and benchmark paths. Preserve meaningful passing criteria.

PR evidence

Link the migration PR here. Report the tested revision, exact commands and results, selected checkpoint cases, original reference checks, current-head CI, and any untested configurations. Keep a normal migration within families/timm_mobilevit/**; raise a separate linked prerequisite issue if a shared contract is genuinely missing. Do not weaken an acceptance criterion to pass CI.

References: #1226 · family validation guide.

Activity

  1. jkzhang7 commented on Oct 1, 2026

    @jkzhang7
    Contributor

    I'd like to work on this migration. Starting now, following the model-validation guide and the timm_mobilenetv3 migration (#1308) as the reference pattern. Will link the PR here once open.

  2. jkzhang7 commented on Oct 1, 2026

    @jkzhang7
    Contributor

    PR: #1561

    Migration is complete and offline-verified (architecture-contract suite, perf-matrix, family unit tests). The C++ side has not been compiled/GPU-verified yet (no CUDA/TensorRT on my development machine) -- disclosed in the PR's Validation section. No inference math changed: model.py's TensorRT graph construction is untouched; only the Task-SDK binding/metadata layer around it changed, closely following #1308's already-merged pattern.

  3. jkzhang7 commented on Oct 6, 2026

    @jkzhang7
    Contributor

    Done via #1561, merged.

  4. jkzhang7 commented on Oct 6, 2026

    @jkzhang7
    Contributor

    Done via #1561, merged.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions