tao-train-bevfusion

安装量: 1.5K
排名: #7682

安装

npx skills add https://github.com/nvidia/skills --skill tao-train-bevfusion

BEVFusion BEVFusion for multi-sensor 3D object detection. Fuses LiDAR point clouds and camera images in bird's-eye-view (BEV) space. Used in autonomous driving for robust 3D perception. Set pretrained backbone paths for Swin image backbone. BEVFusion requires the BEVFusion-specific TAO container nvcr.io/nvidia/tao/tao-toolkit:5.5.0-pyt . The shared TAO PyTorch 7.0 RC image does not package mmdet3d and fails before any BEVFusion action can parse its spec. The model-skill action is named dataset_convert , but the 5.5 container CLI subtask is bevfusion convert -e . Dataclass Schemas Generated TAO Core schemas are packaged in schemas/.schema.json , with schemas/manifest.json listing available actions. Each generated schema also emits references/spec_template_.yaml from the schema top-level default field. AutoML enablement is declared at the model layer in references/skill_info.yaml via automl_enabled . Runnable AutoML still requires schemas/train.schema.json and references/spec_template_train.yaml to exist and parse. Use the packaged train schema for automl_default_parameters , automl_disabled_parameters , defaults, min/max bounds, enums, option weights, math conditions, dependencies, and popular parameters. Do not expect ~/tao-core at runtime; maintainers regenerate schemas/templates before packaging the skill bank. Train Action Policy This model is AutoML-enabled at the model layer. Before handling any train-stage request, read references/skill_info.yaml and resolve the run override from either an explicit automl_policy value or the user's workflow request. Use automl_policy: on by default and only expose on / off in new launch prompts. Treat phrases like "turn off AutoML", "disable AutoML", "no HPO", or "plain training" as automl_policy: off for this run only. When automl_policy: on , automl_enabled: true , and both schemas/train.schema.json and references/spec_template_train.yaml are packaged, route the train action through tao-skill-bank:tao-run-automl by default with this model's skill_dir . Preserve workflow/application overrides for datasets, specs, output directories, GPU/platform settings, parent checkpoints, and automl_policy . Use direct model training only when automl_policy: off or the packaged train schema/template is missing; in the missing-schema case, report that AutoML is enabled but not runnable for this model until schemas are generated. Show more Installs 585 Repository nvidia/skills GitHub Stars 1.9K First Seen Jun 8, 2026 Security Audits Gen Agent Trust Hub Pass Socket Pass Snyk Pass

返回排行榜