Agent skill · NVIDIA
nv-generate-mr-brain-finetune
Used for finetuning NV-Generate-CTMR MR-Brain v1 for T1, T2, FLAIR, SWI, or MRA data from a NIfTI datalist. Not for clinical or production data approval.
What it needs
About 7k tokens when loaded.
What this skill does
NV-Generate-MR-Brain-Finetune Purpose Used for finetuning the NV-Generate-CTMR rflow-mr-brain v1 diffusion UNet from user-supplied T1, T2, FLAIR, SWI, or MRA NIfTI training volumes. Not for clinical interpretation, regulatory use, or approving synthetic data for production training. The wrapper stages the config glue locally and delegates execution to existing upstream scripts: scripts.diffmodelcreatetrainingdata, scripts.diffmodeltrain, and optionally scripts.diffmodelinfer. It does not execute the notebook. Manifest I/O: inputs are datalist and databasedir; outputs are finetunedcheckpoint, optional inferenceoutputs, and resultjson. The underlying training contract is the upstream config/env JSON (the same one driven from cell [10] of traindiffunettutorial.ipynb). The wrapper stages those JSON files for you and exposes the most-tuned fields as CLI flags; the sections below document the fields, their defaults, and how to monitor/tune a run. Instructions Read skillmanifest.yaml before changing arguments, side effects, or validation gates. Run scripts/runmrbrainfinetune.py from the Medical AI Skills repo root. If a host agent exposes runscript, use runscript("scripts/runmrbrainfinetune.py", args=[...]); otherwise run the Bash/Python command below. For a command-shape review, do not install packages, clone repositories, download weights, or start GPU training. Emit only the wrapper command with the supplied datalist, an explicit --data-base-dir, an explicit --output-dir, and the requested modality. When the user explicitly asks for a training-launch command, do not silently replace it with --preflight; include --preflight only for a preflight request. Use --preflight first when checking a new datalist; remove --preflight only when the user explicitly wants to launch GPU finetuning. For a staged preflight input bundle directory, use BUNDLE/preflightdatalist.json as the datalist and BUNDLE/preflightdataset as --data-base-dir when those files are present. …
How to use it
Reference it in AdaL, Claude Code, Cursor or any coding agent — nothing to install:
@skills NVIDIA/nv-generate-mr-brain-finetune